diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 4075c65..1f96ae0 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -1502,8 +1502,7 @@
             break;
 #endif
         default:
-            ret = MBEDTLS_ERR_MD_BAD_INPUT_DATA;
-            break;
+            return( PSA_ERROR_BAD_STATE );
     }
 
     if( ret != 0 )
@@ -1575,8 +1574,7 @@
             break;
 #endif
         default:
-            ret = MBEDTLS_ERR_MD_BAD_INPUT_DATA;
-            break;
+            return( PSA_ERROR_BAD_STATE );
     }
     status = mbedtls_to_psa_error( ret );