Modifying test to match new method signature.