increasing coverage and removing unused import