diff --git a/tests/scripts/check-generated-files.sh b/tests/scripts/check-generated-files.sh
index 2f20026..d829419 100755
--- a/tests/scripts/check-generated-files.sh
+++ b/tests/scripts/check-generated-files.sh
@@ -148,4 +148,4 @@
 # Generated files that are present in the repository even in the development
 # branch. (This is intended to be temporary, until the generator scripts are
 # fully reviewed and the build scripts support a generated header file.)
-check tests/scripts/generate_psa_wrappers.py tests/include/test/psa_test_wrappers.h tests/src/psa_test_wrappers.c
+check tests/scripts/generate_psa_wrappers.py framework/include/test/psa_test_wrappers.h framework/src/psa_test_wrappers.c
diff --git a/tests/scripts/generate_psa_wrappers.py b/tests/scripts/generate_psa_wrappers.py
index 07d1450..1a47e9f 100755
--- a/tests/scripts/generate_psa_wrappers.py
+++ b/tests/scripts/generate_psa_wrappers.py
@@ -225,8 +225,8 @@
         return super()._printf_parameters(typ, var)
 
 
-DEFAULT_C_OUTPUT_FILE_NAME = 'tests/src/psa_test_wrappers.c'
-DEFAULT_H_OUTPUT_FILE_NAME = 'tests/include/test/psa_test_wrappers.h'
+DEFAULT_C_OUTPUT_FILE_NAME = 'framework/src/psa_test_wrappers.c'
+DEFAULT_H_OUTPUT_FILE_NAME = 'framework/include/test/psa_test_wrappers.h'
 
 def main() -> None:
     parser = argparse.ArgumentParser(description=globals()['__doc__'])