fix pyflakes import