diff --git a/test/test.py b/test/test.py
index 58a6eea319c1fa61d5a5ca11f1ecffddbfed6a66..65eaf189853c9f0ffc99fa8b3adf7a8d811c9f07 100644
--- a/test/test.py
+++ b/test/test.py
@@ -185,7 +185,7 @@ class Test:
     def _get_exit_code_from_stdout(self, stdout: str) -> int:
         # Find line with keyword "code"
         for line in stdout.splitlines():
-            if "code" in line:
+            if "exited with code" in line:
                 # Assumes that exit code is after keyword code and that it is the only digits
                 return int(''.join([c for c in line.split('code')[-1] if c.isdigit()]))
         raise AssertionError('No exit code found')