Fix pyflakes compliance