fix flake8 errors for the previous tests