Fix pyflakes linter error