diff --git a/tests/scripts/test_psa_compliance.py b/tests/scripts/test_psa_compliance.py
index bed6d84..2482d03 100755
--- a/tests/scripts/test_psa_compliance.py
+++ b/tests/scripts/test_psa_compliance.py
@@ -52,10 +52,10 @@
 
     if in_tf_psa_crypto_repo:
         crypto_name = 'tfpsacrypto'
-        library_subdir = 'core'
     else:
         crypto_name = 'mbedcrypto'
-        library_subdir = 'library'
+
+    library_subdir = build_tree.crypto_core_directory()
 
     crypto_lib_filename = (library_build_dir + '/' +
                            library_subdir + '/' +