diff --git a/batbot/spectrogram/__init__.py b/batbot/spectrogram/__init__.py
index f1fa0f4..d60485c 100644
--- a/batbot/spectrogram/__init__.py
+++ b/batbot/spectrogram/__init__.py
@@ -15,6 +15,7 @@
 import numpy as np
 import pyastar2d
 import scipy.signal  # Ensure this is at the top with other imports
+import scipy.stats
 import tqdm
 from line_profiler import LineProfiler
 from scipy import ndimage