diff --git a/setup.py b/setup.py
index d3d255e39..10c985035 100644
--- a/setup.py
+++ b/setup.py
@@ -3,9 +3,9 @@
 from Cython.Build import cythonize
 import numpy
 import os
-import imp
+from importlib.machinery import SourceFileLoader
 
-VERSION = imp.load_source('version', os.path.join('.', 'darkflow', 'version.py'))
+VERSION = SourceFileLoader('version', './darkflow/version.py').load_module()
 VERSION = VERSION.__version__
 
 if os.name =='nt' :
@@ -71,4 +71,4 @@
     packages = find_packages(),
 	scripts = ['flow'],
     ext_modules = cythonize(ext_modules)
-)
\ No newline at end of file
+)