fixing flake8 tests