STY: Cleaner imports

Makes the PEP8 checker happier.