Update test to accomdate updated method.