diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index f5a8793..8312c07 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -1002,10 +1002,7 @@
 
     if( slot->lock_count != 1 )
     {
-#if defined(MBEDTLS_TEST_HOOKS)
-        if( *mbedtls_test_hook_assert_test != NULL )
-            ( *mbedtls_test_hook_assert_test )( slot->lock_count == 1, __FILE__, __LINE__ );
-#endif
+        MBEDTLS_TEST_HOOK_TEST_ASSERT( slot->lock_count == 1 );
         status = PSA_ERROR_CORRUPTION_DETECTED;
     }