Fix pyflakes warning
Fix pep8 warning