diff --git a/scripts/generate_driver_wrappers.py b/scripts/generate_driver_wrappers.py
index ec79c4e..7bcac84 100755
--- a/scripts/generate_driver_wrappers.py
+++ b/scripts/generate_driver_wrappers.py
@@ -17,6 +17,8 @@
 import argparse
 import jsonschema
 import jinja2
+
+import framework_path # pylint: disable=unused-import
 from mbedtls_framework import build_tree
 
 JSONSchema = NewType('JSONSchema', object)