update mocks for passing test again