diff --git a/nexus-non-interactive-test.sh b/nexus-non-interactive-test.sh
index d0a23e9..9d4dfa9 100755
--- a/nexus-non-interactive-test.sh
+++ b/nexus-non-interactive-test.sh
@@ -119,7 +119,7 @@
 fi
 
 timeout "${LOG_TIMEOUT}" bash -c '
-    until grep -q -e PASS! -e FAIL! uart.sc.log uart.smc.log ; do
+    until grep -q -e PASS! -e FAIL! -e "= End Benchmark =" uart.sc.log uart.smc.log ; do
         echo "Expected log is missing. Wait up to ${LOG_TIMEOUT}s."
         sleep "${LOG_CHECK_INTERVAL}"
     done' || echo "Time out waiting for PASS! or FAIL! log"
@@ -129,5 +129,5 @@
 cat -n uart.smc.log
 cat -n uart.smc.err
 
-grep -q "PASS!" uart.sc.log uart.smc.log
+grep -q -e "PASS!" -e "= End Benchmark =" uart.sc.log uart.smc.log
 exit $?