diff --git a/setup.py b/setup.py
index 12799fe..19e5463 100644
--- a/setup.py
+++ b/setup.py
@@ -8,7 +8,7 @@
 
 def load_version():
     version_line = open(VERSION_FILE).read().rstrip()
-    vre = re.compile(r'__version__ = "([^"]+)"')
+    vre = re.compile(r'__version__: str = "([^"]+)"')
     matches = vre.findall(version_line)
 
     if matches and len(matches) > 0: