Fix flake8 problem