diff --git a/bnpm/linear_regression.py b/bnpm/linear_regression.py
index 1b4b948..16bd7c9 100644
--- a/bnpm/linear_regression.py
+++ b/bnpm/linear_regression.py
@@ -2,6 +2,8 @@
 import gc
 
 import numpy as np
+import sklearn
+import sklearn.linear_model
 
 
 def LinearRegression_sweep(X_in,