setup.py imports fix