flake8 fixed