diff --git a/library/psa_crypto_pake.c b/library/psa_crypto_pake.c
index 9f91a5e..594def2 100644
--- a/library/psa_crypto_pake.c
+++ b/library/psa_crypto_pake.c
@@ -46,7 +46,7 @@
  *   |   | psa_pake_set_password_key()
  *   |   | psa_pake_set_user()
  *   |   | psa_pake_set_peer()
- *   |   | psa_pake_set_role
+ *   |   | psa_pake_set_role()
  *   |
  *   |--- In any order: (First round input before or after first round output)
  *   |   |
@@ -449,7 +449,7 @@
     }
     else
     {
-        /* Length is stored at the first byte */
+        /* Length is stored at the first byte of the next chunk */
         length = operation->buffer[operation->buffer_offset] + 1;
     }