Fix pyflakes