fix typo, added more expectation for mocks