Fix mock object expectation.