diff --git a/include/psa/crypto_extra.h b/include/psa/crypto_extra.h
index 1d91da01a..a7d98a084 100644
--- a/include/psa/crypto_extra.h
+++ b/include/psa/crypto_extra.h
@@ -2016,9 +2016,9 @@ struct psa_jpake_computation_stage_s {
     psa_jpake_round_t MBEDTLS_PRIVATE(round);
     /* The 'mode' we are currently in (inputting or outputting) */
     psa_jpake_io_mode_t MBEDTLS_PRIVATE(io_mode);
-    /* The number of inputs so far this round */
+    /* The number of completed inputs so far this round */
     uint8_t MBEDTLS_PRIVATE(inputs);
-    /* The number of outputs so far this round */
+    /* The number of completed outputs so far this round */
     uint8_t MBEDTLS_PRIVATE(outputs);
     /* The next expected step (KEY_SHARE, ZK_PUBLIC or ZK_PROOF) */
     psa_pake_step_t MBEDTLS_PRIVATE(step);