setup.py: Import cleanup