diff --git a/mlsauce/elasticnet/__init__.py b/mlsauce/elasticnet/__init__.py
index 2cbe47c..b73aa88 100644
--- a/mlsauce/elasticnet/__init__.py
+++ b/mlsauce/elasticnet/__init__.py
@@ -1,3 +1,6 @@
-from .enet import ElasticNetRegressor
+try:
+    from .enet import ElasticNetRegressor
+except ImportError:
+    pass
 
 __all__ = ["ElasticNetRegressor"]
\ No newline at end of file
diff --git a/setup.py b/setup.py
index d56e1b9..eeb744e 100644
--- a/setup.py
+++ b/setup.py
@@ -177,6 +177,7 @@ def setup_package():
         "numpy",
         "Cython",
         "joblib",
+        "matplotlib",
         "pandas",
         "requests",
         "scikit-learn",