diff --git a/test_mux_2024.py b/test_mux_2024.py
index 9c801dc9e4288cc5a07123d0d2e6fa9e6d6eb74d..49059bb6ec3ee6f30bcdcc2fe16febc228a0e71b 100644
--- a/test_mux_2024.py
+++ b/test_mux_2024.py
@@ -2,8 +2,9 @@ import time
 from utils import change_config
 change_config('config_ohmpi_card_3_15.py', verbose=False)
 from OhmPi.hardware_components.mux_2024_rev_0_0 import Mux, MUX_CONFIG
+from OhmPi.hardware_components import raspberry_pi as controller_module
 
-mux = Mux()
+mux = Mux({'controller': controller_module.Controller()})
 mux.switch_one(elec=1, role='M', state='on')
 time.sleep(2)
 mux.switch_one(elec=1, role='M', state='off')