Fix unused import
PEP8