Fixing build failure: a incorrect method mocking