diff --git a/setup.py b/setup.py
index 9feb1c8..dcc951c 100755
--- a/setup.py
+++ b/setup.py
@@ -1,9 +1,20 @@
 #!/usr/bin/env python
+import importlib.util
+import importlib.machinery
 from setuptools import setup, find_packages
-from imp import load_source
 from os import path
 import io
 
+def load_source(modname, filename):
+    loader = importlib.machinery.SourceFileLoader(modname, filename)
+    spec = importlib.util.spec_from_file_location(modname, filename, loader=loader)
+    module = importlib.util.module_from_spec(spec)
+    # The module is always executed and not cached in sys.modules.
+    # Uncomment the following line to cache the module.
+    # sys.modules[module.__name__] = module
+    loader.exec_module(module)
+    return module
+
 __version__ = load_source('satsearch.version', 'satsearch/version.py').__version__
 
 here = path.abspath(path.dirname(__file__))