fix test fail for pk_sign_ext

Signed-off-by: Jerry Yu <jerry.h.yu@arm.com>
diff --git a/tests/suites/test_suite_pk.function b/tests/suites/test_suite_pk.function
index e89ae74..dc6bf18 100644
--- a/tests/suites/test_suite_pk.function
+++ b/tests/suites/test_suite_pk.function
@@ -1088,7 +1088,7 @@
 }
 /* END_CASE */
 
-/* BEGIN_CASE depends_on:MBEDTLS_USE_PSA_CRYPTO */
+/* BEGIN_CASE depends_on:MBEDTLS_PSA_CRYPTO_C */
 void pk_psa_sign_ext( int pk_type, int parameter, int key_pk_type, int md_alg )
 {
     mbedtls_pk_context pk;
@@ -1101,7 +1101,7 @@
     memset( sig, 0, sizeof sig );
 
     mbedtls_pk_init( &pk );
-    USE_PSA_INIT();
+    PSA_INIT();
 
     TEST_ASSERT( mbedtls_pk_setup( &pk, mbedtls_pk_info_from_type( pk_type ) ) == 0 );
 
@@ -1112,8 +1112,8 @@
                                       mbedtls_test_rnd_std_rand, NULL ) == 0 );
 
 exit:
+    PSA_DONE( );
     mbedtls_pk_free( &pk );
-    USE_PSA_DONE( );
 }
 /* END_CASE */