diff --git a/cocotbext/axi/axis.py b/cocotbext/axi/axis.py
index 2714a7c..8840800 100644
--- a/cocotbext/axi/axis.py
+++ b/cocotbext/axi/axis.py
@@ -28,7 +28,7 @@ from collections import deque
 import cocotb
 from cocotb.triggers import RisingEdge, Timer, First, Event
 from cocotb.utils import get_sim_time
-from cocotb.bus import Bus
+from cocotb_bus.bus import Bus
 
 from .version import __version__
 from .reset import Reset
diff --git a/cocotbext/axi/stream.py b/cocotbext/axi/stream.py
index d673275..de52873 100644
--- a/cocotbext/axi/stream.py
+++ b/cocotbext/axi/stream.py
@@ -27,7 +27,7 @@ from collections import deque
 
 import cocotb
 from cocotb.triggers import RisingEdge, Event, First, Timer
-from cocotb.bus import Bus
+from cocotb_bus.bus import Bus
 
 from .reset import Reset
 
diff --git a/setup.cfg b/setup.cfg
index a27d659..86cde44 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -27,6 +27,7 @@ packages = find_namespace:
 python_requires = >=3.6
 install_requires =
     cocotb
+    cocotb-bus
 
 [options.extras_require]
 test =