diff --git a/experiment_utils/experiment_analyzer.py b/experiment_utils/experiment_analyzer.py
index 20e63d8..14e7f8b 100644
--- a/experiment_utils/experiment_analyzer.py
+++ b/experiment_utils/experiment_analyzer.py
@@ -8,6 +8,7 @@
 import numpy as np
 
 from scipy.stats import gaussian_kde
+from scipy import stats
 from pyspark.sql import functions as F
 from pyspark.sql import DataFrame
 from .utils import log_and_raise_error, get_logger