diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 2c1a910..1c066ce 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -7863,6 +7863,7 @@
 
     status = psa_driver_wrapper_pake_get_implicit_key(operation,
                                                       shared_key,
+                                                      sizeof(shared_key),
                                                       &shared_key_len);
 
     if (status != PSA_SUCCESS) {