Fix pyflakes issue