diff --git a/doc/requirements.txt b/doc/requirements.txt
index 22ed6d5..9da7983 100644
--- a/doc/requirements.txt
+++ b/doc/requirements.txt
@@ -1,7 +1,7 @@
 alabaster==0.7.12
 appdirs==1.4.4
 autograd==1.4
-autoray==0.3.2
+autoray
 Babel==2.10.3
 cachetools==5.2.0
 certifi==2023.7.22
@@ -15,9 +15,9 @@ MarkupSafe==2.1.1
 networkx==2.6.0
 ninja==1.10.2.3
 numpy==1.22.4
-packaging>=24
-PennyLane==0.24.0
-PennyLane-Lightning==0.24.0
+packaging>22.0
+PennyLane
+PennyLane-Lightning
 Pygments==2.15.0
 pyparsing==3.0.9
 python-dateutil==2.8.2
diff --git a/pennylane_ionq/device.py b/pennylane_ionq/device.py
index 6f7cd37..6a7f9b3 100644
--- a/pennylane_ionq/device.py
+++ b/pennylane_ionq/device.py
@@ -19,7 +19,8 @@
 
 import numpy as np
 
-from pennylane import QubitDevice, DeviceError
+from pennylane import DeviceError
+from pennylane.devices import QubitDevice
 
 from .api_client import Job, JobExecutionError
 from ._version import __version__