pep8: Fix trivial missing imports