diff --git a/test_mux_2024.py b/test_mux_2024.py
index f431efd333fe339c6f9490ac5b104218c7d33519..3d76ef3fefeea44983860d2d355dbbd20ca4a4fc 100644
--- a/test_mux_2024.py
+++ b/test_mux_2024.py
@@ -4,8 +4,8 @@ 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
 
-print(MUX_CONFIG)
-mux = Mux({'controller': controller_module.Controller()})
+MUX_CONFIG['controller'] = controller_module.Controller()
+mux = Mux(MUX_CONFIG)
 mux.switch_one(elec=1, role='M', state='on')
 time.sleep(2)
 mux.switch_one(elec=1, role='M', state='off')