Fixed pyflakes