Fixed tests with updated flake8