Fixed pyflakes warning