Fixed missing method in the test.