diff --git a/psasim/README.md b/psasim/README.md
index db49ae947..aaf66e28d 100644
--- a/psasim/README.md
+++ b/psasim/README.md
@@ -34,9 +34,9 @@ A significant portion of the intermediate code of PSASIM is auto-generated using
 These files need to be regenerated whenever some PSA Crypto API is added/deleted/modified. The procedure is as follows:
 
 - `psa_sim_serialise.[c|h]`:
-    - go to `<mbedtls-root>/tests/psa-client-server/psasim/src/`
+    - go to `framework/psasim/src/`
     - run `./psa_sim_serialise.pl h > psa_sim_serialise.h`
     - run `./psa_sim_serialise.pl c > psa_sim_serialise.c`
 - `psa_sim_crypto_[client|server].c` and `psa_functions_codes.h`:
-    - go to Mbed TLS' root folder
-    - run `./tests/psa-client-server/psasim/src/psa_sim_generate.pl`
+    - go to the TF-PSA-Crypto root folder
+    - run `framework/psasim/src/psa_sim_generate.pl`
diff --git a/psasim/src/psa_sim_generate.pl b/psasim/src/psa_sim_generate.pl
index 0f4c86f81..e993f4f22 100755
--- a/psasim/src/psa_sim_generate.pl
+++ b/psasim/src/psa_sim_generate.pl
@@ -1052,7 +1052,7 @@ sub output_signature
 
 sub get_functions
 {
-    my $header_dir = 'tf-psa-crypto/include';
+    my $header_dir = 'include';
     my $src = "";
     for my $header_file ('psa/crypto.h', 'psa/crypto_extra.h') {
         local *HEADER;