diff --git a/setup.py b/setup.py
index cc50448..1082124 100644
--- a/setup.py
+++ b/setup.py
@@ -1,14 +1,6 @@
 from setuptools import setup, find_packages
-import os
 
-version = None
-with open(os.path.join('yaglm', '__init__.py'), 'r') as fid:
-    for line in (line.strip() for line in fid):
-        if line.startswith('__version__'):
-            version = line.split('=')[1].strip().strip('\'')
-            break
-if version is None:
-    raise RuntimeError('Could not determine version')
+from yaglm import __version__ as version
 
 
 install_requires = ['numpy',
@@ -17,7 +9,6 @@
                     'scipy'
                     ]
 
-
 setup(name='yaglm',
       version=version,
       description='A python package for penalized generalized linear models that supports fitting and model selection for structured, adaptive and non-convex penalties.',