Add a mock object for tests