fix import error for setup.py