Removing unused import causing pylint violation