diff --git a/lab5/eval/eval.py b/lab5/eval/eval.py
index 5a20969..4e9f330 100644
--- a/lab5/eval/eval.py
+++ b/lab5/eval/eval.py
@@ -1,6 +1,6 @@
 import csv
 import pandas as pd
-import tensorflow as tf
+from tensorflow.keras.models import load_model
 
 
 X_test = pd.read_csv('test.csv')
@@ -8,7 +8,7 @@ X_test = pd.read_csv('test.csv')
 Y_test = X_test.pop('stabf')
 Y_test = pd.get_dummies(Y_test)
 
-model = tf.keras.models.load_model('grid-stability-dense.h5')
+model = load_model('grid-stability-dense.h5')
 results = model.evaluate(X_test, Y_test, batch_size=64)
 
 with open('eval.csv', 'a', newline='') as fp: