Correctly stub the method in the test