diff --git a/library/CMakeLists.txt b/library/CMakeLists.txt
index 7b7fca4..d7407c3 100644
--- a/library/CMakeLists.txt
+++ b/library/CMakeLists.txt
@@ -164,8 +164,8 @@
         COMMAND
             ${MBEDTLS_PYTHON_EXECUTABLE}
                 ${CMAKE_CURRENT_SOURCE_DIR}/../scripts/generate_driver_wrappers.py
-                ${CMAKE_CURRENT_SOURCE_DIR}/../scripts/data_files/driver_templates/psa_crypto_driver_wrappers.conf
-                ${CMAKE_CURRENT_BINARY_DIR}/psa_crypto_driver_wrappers.c
+                --mbedtls-root ${CMAKE_CURRENT_SOURCE_DIR}/..
+                ${CMAKE_CURRENT_BINARY_DIR}
         DEPENDS
             ${CMAKE_CURRENT_SOURCE_DIR}/../scripts/generate_driver_wrappers.py
             ${CMAKE_CURRENT_SOURCE_DIR}/../scripts/data_files/driver_templates/psa_crypto_driver_wrappers.conf