diff --git a/projects/aca-content/src/lib/components/knowledge-retrieval/search-ai/search-ai-results/search-ai-results.component.spec.ts b/projects/aca-content/src/lib/components/knowledge-retrieval/search-ai/search-ai-results/search-ai-results.component.spec.ts
index f336ea711..9a69317f6 100644
--- a/projects/aca-content/src/lib/components/knowledge-retrieval/search-ai/search-ai-results/search-ai-results.component.spec.ts
+++ b/projects/aca-content/src/lib/components/knowledge-retrieval/search-ai/search-ai-results/search-ai-results.component.spec.ts
@@ -46,7 +46,14 @@ import { searchAiMarkedOptions } from './search-ai-marked-options';
 
 const questionMock: QuestionModel = { question: 'test', questionId: 'testId', restrictionQuery: { nodesIds: [] } };
 const getAiAnswerEntry = (noAnswer?: boolean): AiAnswerEntry => {
-  return { entry: { answer: noAnswer ? '' : 'Some answer', questionId: 'some id', references: [] } };
+  return {
+    entry: {
+      answer: noAnswer ? '' : 'Some answer',
+      question: 'some question',
+      objectReferences: [],
+      complete: true
+    }
+  };
 };
 
 describe('SearchAiResultsComponent', () => {
@@ -512,7 +519,12 @@ describe('SearchAiResultsComponent', () => {
       spyOn(userPreferencesService, 'set');
       spyOn(userPreferencesService, 'get').and.returnValue(knowledgeRetrievalNodes);
       const answer = getAiAnswerEntry();
-      answer.entry.references = [{ referenceId: nodeId, referenceText: 'some text' }];
+      answer.entry.objectReferences = [
+        {
+          objectId: nodeId,
+          references: []
+        }
+      ];
       spyOn(searchAiService, 'getAnswer').and.returnValues(throwError('error'), of(answer));
       mockQueryParams.next({ query: 'test', agentId: 'agentId1' });
 
diff --git a/projects/aca-content/src/lib/components/knowledge-retrieval/search-ai/search-ai-results/search-ai-results.component.ts b/projects/aca-content/src/lib/components/knowledge-retrieval/search-ai/search-ai-results/search-ai-results.component.ts
index 230101cac..92b6404e1 100644
--- a/projects/aca-content/src/lib/components/knowledge-retrieval/search-ai/search-ai-results/search-ai-results.component.ts
+++ b/projects/aca-content/src/lib/components/knowledge-retrieval/search-ai/search-ai-results/search-ai-results.component.ts
@@ -198,7 +198,7 @@ export class SearchAiResultsComponent extends PageComponent implements OnInit {
           }
           this.queryAnswer = response.entry;
           this._displayedAnswer = this.preprocessMarkdownFormat(response.entry.answer);
-          return forkJoin(this.queryAnswer.references.map((reference) => this.nodesApiService.getNode(reference.referenceId)));
+          return forkJoin(this.queryAnswer.objectReferences.map((reference) => this.nodesApiService.getNode(reference.objectId)));
         }),
         retryWhen((errors: Observable<Error>) => this.aiSearchRetryWhen(errors)),
         finalize(() => (this._loading = false)),