diff --git a/scripts/mbedtls_framework/c_build_helper.py b/scripts/mbedtls_framework/c_build_helper.py
index 2081d13..59bb326 100644
--- a/scripts/mbedtls_framework/c_build_helper.py
+++ b/scripts/mbedtls_framework/c_build_helper.py
@@ -16,8 +16,7 @@
 
     def __init__(self, message):
         """Save the error massage"""
-
-        super().__init__()
+        super().__init__(message)
         self.message = message
 
 def remove_file_if_exists(filename):
diff --git a/scripts/mbedtls_framework/psa_storage.py b/scripts/mbedtls_framework/psa_storage.py
index 69e2423..22d981c 100644
--- a/scripts/mbedtls_framework/psa_storage.py
+++ b/scripts/mbedtls_framework/psa_storage.py
@@ -57,6 +57,7 @@
                 includes.append('tf-psa-crypto/drivers/builtin/include')
                 includes.append('tf-psa-crypto/drivers/everest/include')
                 includes.append('tf-psa-crypto/drivers/everest/include/tf-psa-crypto/private/')
+                includes.append('tf-psa-crypto/drivers/pqcp/include')
 
         values = c_build_helper.get_c_expression_values(
             'unsigned long', '%lu',