Fixed import error in setup.py