fix setup.py import