setup.py: fix imports