diff --git a/tests/scripts/generate_psa_wrappers.py b/tests/scripts/generate_psa_wrappers.py
index 1ab71d3..755a584 100755
--- a/tests/scripts/generate_psa_wrappers.py
+++ b/tests/scripts/generate_psa_wrappers.py
@@ -140,10 +140,10 @@
         """
         if not buffer_parameters:
             return
-        out.write('#if defined(MBEDTLS_PSA_COPY_CALLER_BUFFERS)\n')
+        out.write('#if !defined(MBEDTLS_PSA_ASSUME_EXCLUSIVE_BUFFERS)\n')
         for param in buffer_parameters:
             self._write_poison_buffer_parameter(out, param, poison)
-        out.write('#endif /* defined(MBEDTLS_PSA_COPY_CALLER_BUFFERS) */\n')
+        out.write('#endif /* !defined(MBEDTLS_PSA_ASSUME_EXCLUSIVE_BUFFERS) */\n')
 
     @staticmethod
     def _parameter_should_be_copied(function_name: str,