diff --git a/examples/security/tee/tee_secure_ota/pytest_tee_secure_ota.py b/examples/security/tee/tee_secure_ota/pytest_tee_secure_ota.py
index cd201e60d1..508026efd2 100644
--- a/examples/security/tee/tee_secure_ota/pytest_tee_secure_ota.py
+++ b/examples/security/tee/tee_secure_ota/pytest_tee_secure_ota.py
@@ -92,7 +92,7 @@ def test_examples_tee_secure_ota_example(dut: Dut) -> None:
                 raise ValueError('ENV_TEST_FAILURE: Cannot connect to AP')
 
             host_ip = get_host_ip4_by_dest_ip(ip_address)
-            dut.expect('Returned from app_main', timeout=30)
+            dut.expect(f'{dut.target}>', timeout=30)
 
             # User OTA for last iteration
             if i == (iterations - 1):