[fix] Fixed some flake8 warnings