Tiny flake8 fix in import order