DW: corrected the flake8