diff --git a/scripts/data_files/driver_templates/psa_crypto_driver_wrappers.h.jinja b/scripts/data_files/driver_templates/psa_crypto_driver_wrappers.h.jinja
index ac98c19..08d20df 100644
--- a/scripts/data_files/driver_templates/psa_crypto_driver_wrappers.h.jinja
+++ b/scripts/data_files/driver_templates/psa_crypto_driver_wrappers.h.jinja
@@ -2733,11 +2733,6 @@
     size_t output_size,
     size_t *output_length )
 {
-    (void) step;
-    (void) output;
-    (void) output_size;
-    (void) output_length;
-
     switch( operation->id )
     {
 #if defined(MBEDTLS_PSA_BUILTIN_PAKE)
@@ -2755,6 +2750,10 @@
 #endif /* PSA_CRYPTO_DRIVER_TEST */
 #endif /* PSA_CRYPTO_ACCELERATOR_DRIVER_PRESENT */
         default:
+            (void) step;
+            (void) output;
+            (void) output_size;
+            (void) output_length;
             return( PSA_ERROR_INVALID_ARGUMENT );
     }
 }