diff --git a/data/extract.py b/data/extract.py
index eca591cf16e9857218fea1bcf364df3c23fb3213..d384f54483187232986907d3d81f1126c6cc497e 100644
--- a/data/extract.py
+++ b/data/extract.py
@@ -2,6 +2,7 @@ import sys
 import json
 import os
 import distutils.core
+from os.path import dirname, realpath
 
 if sys.version_info[:2] >= (3, 3):
   from importlib.machinery import SourceFileLoader
@@ -32,6 +33,8 @@ except ImportError:
 @mock.patch.object(setuptools, 'setup')
 @mock.patch.object(distutils.core, 'setup')
 def invoke(mock1, mock2):
+  # Inserting the parent directory of the target setup.py in Python import path:
+  sys.path.append(dirname(realpath(sys.argv[-1])))
   # This is setup.py which calls setuptools.setup
   load_source('_target_setup_', sys.argv[-1])
   # called arguments are in `mock_setup.call_args`