Remove unused import

PEP8 up, too.