Remove a unneeded import
PEP8