Update mocks for unit tests