Remove unused import in tests