diff --git a/scripts/generate_psa_constants.py b/scripts/generate_psa_constants.py
index d472c6d..a636363 100755
--- a/scripts/generate_psa_constants.py
+++ b/scripts/generate_psa_constants.py
@@ -1,13 +1,13 @@
 #!/usr/bin/env python3
 
 """Generate psa_constant_names_generated.c
-which is included by programs/psa/psa_constant_names.c.
+which is included by tf-psa-crypto/programs/psa/psa_constant_names.c.
 The code generated by this module is only meant to be used in the context
 of that program.
 
 An argument passed to this script will modify the output directory where the
 file is written:
-* by default (no arguments passed): writes to programs/psa/
+* by default (no arguments passed): writes to tf-psa-crypto/programs/psa/
 * OUTPUT_FILE_DIR passed: writes to OUTPUT_FILE_DIR/
 """
 
@@ -327,7 +327,7 @@
 if __name__ == '__main__':
     build_tree.chdir_to_root()
     # Allow to change the directory where psa_constant_names_generated.c is written to.
-    OUTPUT_FILE_DIR = sys.argv[1] if len(sys.argv) == 2 else "programs/psa"
+    OUTPUT_FILE_DIR = sys.argv[1] if len(sys.argv) == 2 else "tf-psa-crypto/programs/psa"
 
     generate_psa_constants(['tf-psa-crypto/include/psa/crypto_values.h',
                             'tf-psa-crypto/include/psa/crypto_extra.h'],