diff --git a/crit/setup.py b/crit/setup.py
index 1aaa73a13..2f584678f 100644
--- a/crit/setup.py
+++ b/crit/setup.py
@@ -1,23 +1,9 @@
-import os
 from setuptools import setup, find_packages
-
-
-def get_version():
-    version = '0.0.1'
-    env = os.environ
-    if 'CRIU_VERSION_MAJOR' in env and 'CRIU_VERSION_MINOR' in env:
-        version = '{}.{}'.format(
-            env['CRIU_VERSION_MAJOR'],
-            env['CRIU_VERSION_MINOR']
-        )
-        if 'CRIU_VERSION_SUBLEVEL' in env and env['CRIU_VERSION_SUBLEVEL']:
-            version += '.' + env['CRIU_VERSION_SUBLEVEL']
-    return version
-
+import pycriu
 
 setup(
     name='crit',
-    version=get_version(),
+    version=pycriu.__version__,
     description='CRiu Image Tool',
     author='CRIU team',
     author_email='criu@openvz.org',