fix for flake8 check