diff --git a/measure.py b/measure.py
index 3ec42d481f49da496d3b12637b84dcca2944b55a..10b824025ba7d78fcf270851772145e20b0258fe 100644
--- a/measure.py
+++ b/measure.py
@@ -153,7 +153,7 @@ class OhmPiHardware:
 
     def vab_square_wave(self, vab, length, sampling_rate, cycles=3, polarity=1):
         self.tx.polarity = polarity
-        self._vab_pulses(vab, [length]*cycles, sampling_rate)
+        self._vab_pulses(vab, [length]*2*cycles, sampling_rate)
 
     def _vab_pulse(self, vab, length, sampling_rate=None, polarity=None, append=False):
         """ Gets VMN and IAB from a single voltage pulse
@@ -183,5 +183,6 @@ class OhmPiHardware:
         else:
             polarities = [-self.tx.polarity * np.heaviside(i % 2, -1.) for i in range(n_pulses)]
         self._clear_values()
+        print(f'Polarities: {polarities}') # TODO: delete me
         for i in range(n_pulses):
             self._vab_pulse(self, length=lengths[i], sampling_rate=sampling_rate, polarity=polarities[i], append=True)
\ No newline at end of file