Removed unused import which make tests fails