Fix tests mock object