diff --git a/lab5/train/train.py b/lab5/train/train.py
index 21289d2..e9a900c 100644
--- a/lab5/train/train.py
+++ b/lab5/train/train.py
@@ -1,7 +1,7 @@
 from datetime import datetime
 import pandas as pd
 from sacred import Experiment
-from sacred.observers import MongoObserver
+from sacred.observers import FileStorageObserver, MongoObserver
 import sys
 import tensorflow
 from tensorflow.keras import layers