flake8 error fixed