diff --git a/tests/suites/test_suite_pk.function b/tests/suites/test_suite_pk.function
index edd5e66..d5c7d08 100644
--- a/tests/suites/test_suite_pk.function
+++ b/tests/suites/test_suite_pk.function
@@ -438,9 +438,7 @@
     void *options;
     size_t hash_len;
 
-#if defined(MBEDTLS_USE_PSA_CRYPTO)
-    psa_crypto_init();
-#endif
+    USE_PSA_INIT( );
     mbedtls_pk_init( &pk );
 
     memset( hash_result, 0x00, sizeof( hash_result ) );
@@ -484,10 +482,7 @@
 
 exit:
     mbedtls_pk_free( &pk );
-#if defined(MBEDTLS_USE_PSA_CRYPTO)
-    mbedtls_test_psa_purge_key_storage();
-    mbedtls_psa_crypto_free();
-#endif
+    USE_PSA_DONE( );
 }
 /* END_CASE */