diff --git a/tests/src/psa_exercise_key.c b/tests/src/psa_exercise_key.c
index 5aed683ebc..937bd45d22 100644
--- a/tests/src/psa_exercise_key.c
+++ b/tests/src/psa_exercise_key.c
@@ -715,6 +715,7 @@ psa_status_t mbedtls_test_psa_raw_key_agreement_with_self(
         status = PSA_SUCCESS;
         goto exit;
     }
+    PSA_ASSERT(status);
 
     status = psa_raw_key_agreement(alg, key,
                                    public_key, public_key_length,