diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 7f5a7f8..1cb00c8 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -5284,6 +5284,8 @@
         return( PSA_ERROR_INVALID_ARGUMENT );
 
     uint8_t *pms = mbedtls_calloc( 1, pms_len );
+    if( pms == NULL )
+        return( PSA_ERROR_INSUFFICIENT_MEMORY );
     uint8_t *cur = pms;
 
     /* pure-PSK: