diff --git a/examples/provisioning/wifi_prov_mgr/pytest_wifi_prov_mgr.py b/examples/provisioning/wifi_prov_mgr/pytest_wifi_prov_mgr.py
index 71f63436d8..4d5ea37a9c 100644
--- a/examples/provisioning/wifi_prov_mgr/pytest_wifi_prov_mgr.py
+++ b/examples/provisioning/wifi_prov_mgr/pytest_wifi_prov_mgr.py
@@ -6,8 +6,15 @@
 from __future__ import print_function
 
 import logging
+import os
+import sys
+
+try:
+    import esp_prov
+except ImportError:
+    sys.path.append(os.path.join(os.path.dirname(__file__), '..', '..', '..', 'tools'))
+    import esp_prov
 
-import esp_prov
 import pytest
 from pytest_embedded import Dut