Fix some pyflakes errors