diff --git a/advanced_ML/model_tree/models/logistic_regr.py b/advanced_ML/model_tree/models/logistic_regr.py
index 86f4518..4a7a3bb 100644
--- a/advanced_ML/model_tree/models/logistic_regr.py
+++ b/advanced_ML/model_tree/models/logistic_regr.py
@@ -4,6 +4,8 @@
 
 """
 from sklearn.metrics import mean_squared_error
+import numpy as np
+
 
 class logistic_regr: