diff --git a/ohmpi/hardware_system.py b/ohmpi/hardware_system.py
index 801248f1afff91516ed9fe23d1346177d1203d27..2c7a645dfd0f073896d115d4a76dc4eedddf747d 100644
--- a/ohmpi/hardware_system.py
+++ b/ohmpi/hardware_system.py
@@ -692,10 +692,10 @@ class OhmPiHardware:
         for i in range(n_pulses):
             self._vab_pulse(vab=vab, duration=durations[i], sampling_rate=sampling_rate, polarity=polarities[i],
                             append=True)
-        if switch_pwr_off:
-            self.tx.pwr.pwr_state = 'off'
-        # if switch_tx_pwr_off:
-        self.pwr_state = 'off'
+        # if switch_pwr_off:
+        self.tx.pwr.pwr_state = 'off'
+        if switch_tx_pwr_off:
+            self.pwr_state = 'off'
     
     def switch_mux(self, electrodes, roles=None, state='off', **kwargs):
         """Switches on multiplexer relays for given quadrupole.