Fix the import order in setup.py