Fixed some imports (pyflakes)