diff --git a/electrode_check.py b/electrode_check.py
index e7c457f..68c18a5 100644
--- a/electrode_check.py
+++ b/electrode_check.py
@@ -360,7 +360,7 @@ def main():
                 power_channel = int(np.argmax(channel_std))
 
                 if init_fig == True:
-                    yspan = (np.min(channel_std[power_channel]), np.max(channel_std[power_channel]))
+                    yspan = (np.min(channel_data[power_channel]), np.max(channel_data[power_channel]))
                     ylim = (yspan[0] - np.abs(np.diff(yspan)) * 0.2, yspan[1] + np.abs(np.diff(yspan)) * 0.2)
 
                     for ch in channel_array:
@@ -372,8 +372,8 @@ def main():
 
                     init_fig = False
                 else:
-                    yspan = (np.min(channel_std[power_channel]), np.max(channel_std[power_channel]))
-                    ylim = (yspan[0] - np.abs(np.diff(yspan)) * 0.2, yspan[1] + np.abs(np.diff(yspan)) * 0.2)
+                    yspan = [np.min(channel_data[power_channel]), np.max(channel_data[power_channel])]
+                    ylim = [yspan[0] - np.abs(np.diff(yspan)) * 0.2, yspan[1] + np.abs(np.diff(yspan)) * 0.2]
                     for ch in channel_array:
                         Plot.channel_handle[ch].set_data(np.arange(250)[:len(channel_data[ch])] / rate, channel_data[ch])
                         Plot.axs[ch].set_ylim(ylim)