fix import for setup.py