Fixed failing pyflakes