setup.py: clean up imports