Fix version import in setup.py