diff --git a/test_mux_2024.py b/test_mux_2024.py
index dc0e48b6f13d25a1cd5d2a0f05cbaf9ffdd8dfed..657e0bc7da2cfa3c24f5818465b697d402d5e405 100644
--- a/test_mux_2024.py
+++ b/test_mux_2024.py
@@ -9,5 +9,7 @@ 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')
-mux.switch({'A': [1], 'B': [4], 'M': [2], 'N': [3]})
+mux.switch({'A': [1], 'B': [4], 'M': [2], 'N': [3]}, state='on')
+time.sleep(4)
+mux.switch({'A': [1], 'B': [4], 'M': [2], 'N': [3]}, state='off')
 change_config('config_default.py', verbose=False)