diff --git a/library/psa_crypto.c b/library/psa_crypto.c
index bafb0ae0b4..bfbd497352 100644
--- a/library/psa_crypto.c
+++ b/library/psa_crypto.c
@@ -7247,8 +7247,6 @@ psa_status_t psa_pake_setup(
 
     if (cipher_suite == NULL ||
         PSA_ALG_IS_PAKE(cipher_suite->algorithm) == 0 ||
-        (cipher_suite->type != PSA_PAKE_PRIMITIVE_TYPE_ECC &&
-         cipher_suite->type != PSA_PAKE_PRIMITIVE_TYPE_DH) ||
         PSA_ALG_IS_HASH(cipher_suite->hash) == 0) {
         return PSA_ERROR_INVALID_ARGUMENT;
     }