diff --git a/dev/test_mb_2024_0_mux_2024.py b/dev/test_mb_2024_0_mux_2024.py
index de2995c7b2f6746bab53e800867e7f51a28194d3..1b2270d818a03ad6d05894700aea7266e2ab533a 100644
--- a/dev/test_mb_2024_0_mux_2024.py
+++ b/dev/test_mb_2024_0_mux_2024.py
@@ -54,7 +54,7 @@ if part_of_hardware_system:
     k.switch_mux(electrodes=[1, 4, 2, 3], roles=['A', 'B', 'M', 'N'], state='off')
 
 if within_ohmpi:
-    from ohmpi import OhmPi
+    from ohmpi.ohmpi import OhmPi
     # from ohmpi.plots import plot_exec_log
 
     print('Starting test of mux within OhmPi.')