Fixed flake8 error