Fix a bug
flake8