Fix flake8 warnring