diff --git a/tests/test_first_run.py b/tests/test_first_run.py
index 3444e48..b858f3d 100644
--- a/tests/test_first_run.py
+++ b/tests/test_first_run.py
@@ -25,4 +25,4 @@ def test_training():
     """[summary]
     Run 10 iterations of dummy training and see if it works.
     """
-    assert init_engine('train', ["config=first_run_test/default.yaml"]).run()
+    assert init_engine('train', ["config=first_run_test/default.yaml"]).run() is None