diff --git a/lab5/train/train.py b/lab5/train/train.py
index b8ca4c9..1f3faeb 100644
--- a/lab5/train/train.py
+++ b/lab5/train/train.py
@@ -46,7 +46,7 @@ def prepare_train_model(learning_rate, _run):
 
     model.save('grid-stability-dense.h5')
 
-    _run['history'] = str(history.history[:, -1])
+    _run['history'] = str(history.history['loss'][1])
 
 
 @ex.main