Remove unused import in test