Fixing flake8 tests for travis