diff --git a/test_bluesky.py b/test_bluesky.py
index 66fda57..c378e66 100644
--- a/test_bluesky.py
+++ b/test_bluesky.py
@@ -13,7 +13,6 @@
 from genie_python import genie as g  # type: ignore
 from ibex_bluesky_core.callbacks import ISISCallbacks
 from ibex_bluesky_core.callbacks.fitting.fitting_utils import Linear
-from ibex_bluesky_core.devices import get_pv_prefix
 from ibex_bluesky_core.devices.block import block_r, block_rw_rbv
 from ibex_bluesky_core.devices.simpledae import SimpleDae
 from ibex_bluesky_core.devices.simpledae.controllers import (
@@ -30,6 +29,7 @@
 )
 from ibex_bluesky_core.log import set_bluesky_log_levels
 from ibex_bluesky_core.run_engine import get_run_engine
+from ibex_bluesky_core.utils import get_pv_prefix
 from ophyd_async.plan_stubs import ensure_connected
 
 from utilities.utilities import (