diff --git a/setup.py b/setup.py
index bcc6ac6e..7bd668b5 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,8 @@
 #!/usr/bin/env python
 from setuptools import find_packages
 from distutils.core import setup
-
+import os
+import re
 
 this_directory = os.path.abspath(os.path.dirname(__file__))
 with open(os.path.join(this_directory, 'README.md')) as f: