diff --git a/test_mux_2024.py b/test_mux_2024.py
index ec73a51fe695a42a06d0fc33700bc28a35dcd3e7..ad68ca2c2fe7a99d287877da834f67a54ea71b3d 100644
--- a/test_mux_2024.py
+++ b/test_mux_2024.py
@@ -38,7 +38,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
     print('Starting test of mux within OhmPi.')
     k = OhmPi()
     k.switch_mux((1,'A'), state='on')