diff --git a/simulations/airshower_beacon_simulation/ac_show_signal_to_noise.py b/simulations/airshower_beacon_simulation/ac_show_signal_to_noise.py
index 5c60cfe..5c0d73c 100755
--- a/simulations/airshower_beacon_simulation/ac_show_signal_to_noise.py
+++ b/simulations/airshower_beacon_simulation/ac_show_signal_to_noise.py
@@ -70,14 +70,14 @@ if __name__ == "__main__":
     ## Beacon vs Noise SNR
     ##
     if True:
-        beacon_snrs = [ lib.signal_to_noise(myfilter(beacon_amp*ant.beacon), myfilter(ant.noise), samplerate=1/dt, signal_band=beacon_pb) for ant in antennas ]
+        beacon_snrs = [ lib.signal_to_noise(myfilter(beacon_amp*ant.beacon), myfilter(ant.noise), samplerate=1/dt, signal_band=beacon_pb, noise_band=pb) for ant in antennas ]
 
         fig, ax = plt.subplots()
         ax.set_title("Maximum Beacon/Noise SNR")
         ax.set_xlabel("Antenna no.")
         ax.set_ylabel("SNR")
         ax.plot([ int(ant.name) for ant in antennas], beacon_snrs, 'o', ls='none')
-        
+
         if fig_dir:
             fig.savefig(path.join(fig_dir, path.basename(__file__) + f".beacon_vs_noise_snr.pdf"))
 
@@ -85,7 +85,7 @@ if __name__ == "__main__":
     ## Beacon vs Total SNR
     ##
     if True:
-        beacon_snrs = [ lib.signal_to_noise(myfilter(beacon_amp*ant.beacon), ant.E_AxB, samplerate=1/dt, signal_band=beacon_pb) for ant in antennas ]
+        beacon_snrs = [ lib.signal_to_noise(myfilter(beacon_amp*ant.beacon), ant.E_AxB, samplerate=1/dt, signal_band=beacon_pb, noise_band=pb) for ant in antennas ]
 
         fig, ax = plt.subplots()
         ax.set_title("Maximum Beacon/Total SNR")
@@ -99,16 +99,16 @@ if __name__ == "__main__":
     ## Airshower signal vs Noise SNR
     ##
     if True:
-        shower_snrs = [ lib.signal_to_noise(ant.E_AxB, myfilter(ant.noise), samplerate=1/dt, signal_band=pb) for ant in antennas ]
+        shower_snrs = [ lib.signal_to_noise(ant.E_AxB, myfilter(ant.noise), samplerate=1/dt, signal_band=pb, noise_band=pb) for ant in antennas ]
 
         fig, ax = plt.subplots()
         ax.set_title("Total (Signal+Beacon+Noise)/Noise SNR")
         ax.set_xlabel("Antenna no.")
         ax.set_ylabel("SNR")
         ax.plot([ int(ant.name) for ant in antennas], shower_snrs, 'o', ls='none')
-        
+
         if fig_dir:
             fig.savefig(path.join(fig_dir, path.basename(__file__) + f".total_snr.pdf"))
-    
+
     if show_plots:
         plt.show()