Fixed bug removing unused import