setup.py import fix