Fixed pyflakes import warning