diff --git a/setup.py b/setup.py
index 8f9d8b8aa29611aaa75bec42b51de77cc19c443d..bf45ba93621841d077d6e63c0de7f5124184bb64 100644
--- a/setup.py
+++ b/setup.py
@@ -1,9 +1,10 @@
 #!/usr/bin/python3
 # -*- coding: utf-8 -*-
 #-------------------------------------------------------------------------------
+# This file is part of idea-format package (https://pypi.python.org/pypi/idea-format).
+#
 # Copyright (c) since 2016, CESNET, z. s. p. o.
-# Authors: Pavel Kácha <pavel.kacha@cesnet.cz>
-#          Jan Mach <jan.mach@cesnet.cz>
+# Author: Pavel Kácha <pavel.kacha@cesnet.cz>
 # Use of this source is governed by an ISC license, see LICENSE file.
 #-------------------------------------------------------------------------------
 
@@ -11,23 +12,31 @@
 #   https://packaging.python.org/distributing/
 #   http://python-packaging-user-guide.readthedocs.io/distributing/
 
+import sys
+import os
+
 # Always prefer setuptools over distutils
 from setuptools import setup, find_packages
 # To use a consistent encoding
 from codecs import open
-from os import path
 
+#
+# Import local version of typedcols library, so that we can insert correct version
+# number into documentation.
+#
+sys.path.insert(0, os.path.abspath('.'))
 import idea
 
-here = path.abspath(path.dirname(__file__))
+#-------------------------------------------------------------------------------
+
+here = os.path.abspath(os.path.dirname(__file__))
 
 # Get the long description from the README file
-with open(path.join(here, 'README.rst'), encoding='utf-8') as f:
+with open(os.path.join(here, 'README.rst'), encoding='utf-8') as f:
     long_description = f.read()
 
 setup(
     name = 'idea-format',
-    #version = '0.1.8',
     version = idea.__version__,
     description = 'Python library for working with IDEA messages.',
     long_description = long_description,