Fix pyflakes error