Fix method call & test