Fixed setup.py import problem