Use operation alg for locking key slot

Signed-off-by: Przemek Stekiel <przemyslaw.stekiel@mobica.com>
diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 2d1c065..5e567ad 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -7281,7 +7281,7 @@
 
     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;
     }