Fixed test failure with changes to method signature