diff --git a/ohmpi/hardware_system.py b/ohmpi/hardware_system.py
index 823a56290b5098d2198a47d5b0ddfd3646e373fe..0307270c363dd980acb16ebc5ea5e37c036c25fc 100644
--- a/ohmpi/hardware_system.py
+++ b/ohmpi/hardware_system.py
@@ -456,6 +456,7 @@ class OhmPiHardware:
             #     self.tx.pwr.pwr_state = 'off'
         else:
             vab_opt = tx_volt
+
         # if strategy == 'vmax':
         #     # implement different strategies
         #     if vab < vab_max and iab < current_max:
diff --git a/ohmpi/ohmpi.py b/ohmpi/ohmpi.py
index e11f974fce28182ff0ae868c9f48f49c2b8188bf..59e72b252f8263dd58fe1d641c0d36e5d9748fc7 100644
--- a/ohmpi/ohmpi.py
+++ b/ohmpi/ohmpi.py
@@ -518,6 +518,9 @@ class OhmPi(object):
         d = {}
         if self.switch_mux_on(quad, bypass_check=bypass_check, cmd_id=cmd_id):
             tx_volt = self._hw._compute_tx_volt(tx_volt=tx_volt, strategy=strategy, vmn_max=vmn_max)  # TODO: use tx_volt and vmn_max instead of hardcoded values
+            self._hw.switch_mux(electrodes=quad[0:2], roles=['A', 'B'], state='on')
+            time.sleep(0.5)
+            self._hw.switch_mux(electrodes=quad[0:2], roles=['A', 'B'], state='off')
             self._hw.vab_square_wave(tx_volt, cycle_duration=injection_duration*2/duty_cycle, cycles=nb_stack, duty_cycle=duty_cycle)
             if 'delay' in kwargs.keys():
                 delay = kwargs['delay']
@@ -579,7 +582,6 @@ class OhmPi(object):
             self._hw.switch_mux(electrodes=quad[0:2], roles=['A', 'B'], state='on')
             time.sleep(1.0)
             self._hw.switch_mux(electrodes=quad[0:2], roles=['A', 'B'], state='off')
-
         else:
             self.exec_logger.info(f'Skipping {quad}')
         self.switch_mux_off(quad, cmd_id)