Fixed setup.py import error