Fix flake8

- Fix flake8