flake8-ed the tests