diff --git a/include/psa/crypto_extra.h b/include/psa/crypto_extra.h
index a1792fe..9f46c5f 100644
--- a/include/psa/crypto_extra.h
+++ b/include/psa/crypto_extra.h
@@ -1071,7 +1071,7 @@
     uint8_t *MBEDTLS_PRIVATE(peer);
     size_t MBEDTLS_PRIVATE(peer_len);
     psa_key_attributes_t MBEDTLS_PRIVATE(attributes);
-    psa_pake_cipher_suite_t MBEDTLS_PRIVATE(cipher_suite);
+    struct psa_pake_cipher_suite_s MBEDTLS_PRIVATE(cipher_suite);
 };
 
 typedef enum psa_crypto_driver_pake_step {
@@ -1141,12 +1141,12 @@
     union {
         uint8_t MBEDTLS_PRIVATE(dummy);
 #if defined(PSA_WANT_ALG_JPAKE)
-        psa_jpake_computation_stage_t MBEDTLS_PRIVATE(jpake);
+        struct psa_jpake_computation_stage_s MBEDTLS_PRIVATE(jpake);
 #endif
     } MBEDTLS_PRIVATE(computation_stage);
     union {
         psa_driver_pake_context_t MBEDTLS_PRIVATE(ctx);
-        psa_crypto_driver_pake_inputs_t MBEDTLS_PRIVATE(inputs);
+        struct psa_crypto_driver_pake_inputs_s MBEDTLS_PRIVATE(inputs);
     } MBEDTLS_PRIVATE(data);
 #endif
 };