diff --git a/tests/suites/test_suite_pk.function b/tests/suites/test_suite_pk.function
index 62682256d1..2c08a5e799 100644
--- a/tests/suites/test_suite_pk.function
+++ b/tests/suites/test_suite_pk.function
@@ -303,7 +303,6 @@ exit:
 void pk_invalid_param()
 {
     mbedtls_pk_context ctx;
-    mbedtls_md_type_t md_alg_none = MBEDTLS_MD_NONE;
     mbedtls_pk_type_t pk_type = 0;
     unsigned char buf[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 };
     size_t buf_size = sizeof( buf );
@@ -311,21 +310,37 @@ void pk_invalid_param()
     mbedtls_pk_init( &ctx );
 
     TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
-                mbedtls_pk_verify_restartable( &ctx, md_alg_none,
+                mbedtls_pk_verify_restartable( &ctx, MBEDTLS_MD_NONE,
                                                NULL, buf_size,
                                                buf, buf_size,
                                                NULL ) );
+    TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
+                mbedtls_pk_verify_restartable( &ctx, MBEDTLS_MD_SHA256,
+                                               NULL, 0,
+                                               buf, buf_size,
+                                               NULL ) );
     TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
                 mbedtls_pk_verify_ext( pk_type, NULL,
-                                       &ctx, md_alg_none,
+                                       &ctx, MBEDTLS_MD_NONE,
                                        NULL, buf_size,
                                        buf, buf_size ) );
     TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
-                mbedtls_pk_sign_restartable( &ctx, md_alg_none,
+                mbedtls_pk_verify_ext( pk_type, NULL,
+                                       &ctx, MBEDTLS_MD_SHA256,
+                                       NULL, 0,
+                                       buf, buf_size ) );
+    TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
+                mbedtls_pk_sign_restartable( &ctx, MBEDTLS_MD_NONE,
                                              NULL, buf_size,
                                              buf, buf_size, &buf_size,
                                              NULL, NULL,
                                              NULL ) );
+    TEST_EQUAL( MBEDTLS_ERR_PK_BAD_INPUT_DATA,
+                mbedtls_pk_sign_restartable( &ctx, MBEDTLS_MD_SHA256,
+                                             NULL, 0,
+                                             buf, buf_size, &buf_size,
+                                             NULL, NULL,
+                                             NULL ) );
 exit:
     mbedtls_pk_free( &ctx );
 }