removed unused import
fixed test