Updated the test to new method signatures.