diff --git a/tests/src/drivers/test_driver_pake.c b/tests/src/drivers/test_driver_pake.c
index 06168a1..437c499 100644
--- a/tests/src/drivers/test_driver_pake.c
+++ b/tests/src/drivers/test_driver_pake.c
@@ -39,9 +39,9 @@
 {
     mbedtls_test_driver_pake_hooks.hits++;
 
-    if (mbedtls_test_driver_pake_hooks.forced_status != PSA_SUCCESS) {
+    if (mbedtls_test_driver_pake_hooks.forced_setup_status != PSA_SUCCESS) {
         mbedtls_test_driver_pake_hooks.driver_status =
-            mbedtls_test_driver_pake_hooks.forced_status;
+            mbedtls_test_driver_pake_hooks.forced_setup_status;
     } else {
 #if defined(MBEDTLS_TEST_LIBTESTDRIVER1) && \
         defined(LIBTESTDRIVER1_MBEDTLS_PSA_BUILTIN_PAKE)