diff --git a/library/psa_crypto_pake.c b/library/psa_crypto_pake.c
index 1aa8a72..e7037ce 100644
--- a/library/psa_crypto_pake.c
+++ b/library/psa_crypto_pake.c
@@ -118,7 +118,7 @@
                              const psa_pake_cipher_suite_t *cipher_suite)
 {
     /* A context must be freshly initialized before it can be set up. */
-    if( operation->alg != 0 || operation->state != PSA_PAKE_STATE_INVALID )
+    if( operation->alg != 0 )
         return( PSA_ERROR_BAD_STATE );
 
     if( cipher_suite == NULL ||