diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 2d1c06500..5e567ad12 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -7281,7 +7281,7 @@ psa_status_t psa_pake_set_password_key(
 
     status = psa_get_and_lock_key_slot_with_policy(password, &slot,
                                                    PSA_KEY_USAGE_DERIVE,
-                                                   PSA_ALG_JPAKE);
+                                                   operation->alg);
     if (status != PSA_SUCCESS) {
         return status;
     }