diff --git a/ohmpi/hardware_system.py b/ohmpi/hardware_system.py
index 283001d5369f08114d7a46e70df2238e492c7ecc..bf862fb2d329295dd79083021f4e2ee065480cc1 100644
--- a/ohmpi/hardware_system.py
+++ b/ohmpi/hardware_system.py
@@ -602,6 +602,7 @@ class OhmPiHardware:
         # if self.tx.pwr.pwr_state == 'off':
         #     self.tx.pwr.pwr_state = 'on'
         #     switch_pwr_off = True
+
         self._gain_auto(vab=vab)
         assert 0. <= duty_cycle <= 1.
         if duty_cycle < 1.:
diff --git a/ohmpi/ohmpi.py b/ohmpi/ohmpi.py
index c9df403fc5cce9ec61c05899ba0473979c658638..a20908f19a53c441400c99645965666dbbba7110 100644
--- a/ohmpi/ohmpi.py
+++ b/ohmpi/ohmpi.py
@@ -491,6 +491,7 @@ class OhmPi(object):
         # tx_volt = float(kwargs.pop('tx_volt', self.settings['tx_volt']))
         bypass_check = kwargs['bypass_check'] if 'bypass_check' in kwargs.keys() else False
         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)
             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']