diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index 2b9c8a2..d8ccb8a 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -7960,7 +7960,7 @@
     } else {
         return PSA_JPAKE_STEP_INVALID;
     }
-    return key_share_step + stage->step - PSA_PAKE_STEP_KEY_SHARE;
+    return (psa_crypto_driver_pake_step_t) (key_share_step + stage->step - PSA_PAKE_STEP_KEY_SHARE);
 }
 #endif /* PSA_WANT_ALG_JPAKE */