Fix method name in test and remove unused import