Amend test following method signature change