Fix bug in distribute_arguments for multi-argument macros
diff --git a/tests/scripts/test_psa_constant_names.py b/tests/scripts/test_psa_constant_names.py
index 2021293..9c674e5 100755
--- a/tests/scripts/test_psa_constant_names.py
+++ b/tests/scripts/test_psa_constant_names.py
@@ -105,7 +105,7 @@
                 for value in argument_lists[i][1:]:
                     arguments[i] = value
                     yield self.format_arguments(name, arguments)
-                arguments[i] = argument_lists[0]
+                arguments[i] = argument_lists[0][0]
         except BaseException as e:
             raise Exception('distribute_arguments({})'.format(name)) from e