diff --git a/simple_regression_lab8.py b/simple_regression_lab8.py
index d02e79b..d68e92a 100644
--- a/simple_regression_lab8.py
+++ b/simple_regression_lab8.py
@@ -8,6 +8,7 @@ from sacred import Experiment
 from sacred.observers import FileStorageObserver
 from sacred.observers import MongoObserver
 import mlflow
+import mlflow.keras
 from urllib.parse import urlparse
 
 # Konfiguracja serwera i nazwy eksperymentu MLflow