X-Git-Url: https://zdv2.bktei.com/gitweb/EVA-2020-02-2.git/blobdiff_plain/4d6b7f96ada71d5f303218e84765d9e74583ef70..4cbbc360566337406c8bf7b1d3b653d19b41ae4c:/library/enviroplus/noise.py

diff --git a/library/enviroplus/noise.py b/library/enviroplus/noise.py
index 2e7472d..7b6d5e2 100644
--- a/library/enviroplus/noise.py
+++ b/library/enviroplus/noise.py
@@ -73,10 +73,10 @@ class Noise():
         high_start = mid_start + int(sample_count * mid)
         noise_ceiling = high_start + int(sample_count * high)
 
-        amp_low = numpy.mean(magnitude[self.noise_floor:mid_start])
+        amp_low = numpy.mean(magnitude[noise_floor:mid_start])
         amp_mid = numpy.mean(magnitude[mid_start:high_start])
         amp_high = numpy.mean(magnitude[high_start:noise_ceiling])
-        amp_total = (low + mid + high) / 3.0
+        amp_total = (amp_low + amp_mid + amp_high) / 3.0
 
         return amp_low, amp_mid, amp_high, amp_total