diff --git a/train.py b/train.py
index 90eddff..64d80a9 100644
--- a/train.py
+++ b/train.py
@@ -57,7 +57,7 @@ def train_model(data_file, model_file, epochs, batch_size, test_size, random_sta
     model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
 
     model.fit(X_train, y_train, epochs=epochs, batch_size=batch_size)
-    loss, accuracy = model.evaluate(X_test, y_test)
+    accuracy = model.evaluate(X_test, y_test)
     print('Test accuracy:', accuracy)
 
     model.save(model_file)