diff --git a/library/psa_crypto_pake.c b/library/psa_crypto_pake.c
index 7f147b7..48995dd 100644
--- a/library/psa_crypto_pake.c
+++ b/library/psa_crypto_pake.c
@@ -702,10 +702,6 @@
         return( PSA_SUCCESS );
     }
 
-    operation->alg = 0;
-    operation->state = 0;
-    operation->sequence = 0;
-
 #if defined(MBEDTLS_PSA_BUILTIN_ALG_ECJPAKE)
     if( operation->alg == PSA_ALG_JPAKE )
     {
@@ -721,6 +717,10 @@
     }
 #endif
 
+    operation->alg = 0;
+    operation->state = 0;
+    operation->sequence = 0;
+
     return( PSA_SUCCESS );
 }