diff --git a/packages/editor-ui/src/api/testDefinition.ee.ts b/packages/editor-ui/src/api/testDefinition.ee.ts
index c515b7a783f9f..5f4ce66ef91af 100644
--- a/packages/editor-ui/src/api/testDefinition.ee.ts
+++ b/packages/editor-ui/src/api/testDefinition.ee.ts
@@ -11,7 +11,7 @@ export interface TestDefinitionRecord {
 	updatedAt?: string;
 	createdAt?: string;
 	annotationTag?: string | null;
-	mockedNodes?: Array<{ name: string }>;
+	mockedNodes?: Array<{ name: string; id: string }>;
 }
 
 interface CreateTestDefinitionParams {
@@ -25,7 +25,7 @@ export interface UpdateTestDefinitionParams {
 	evaluationWorkflowId?: string | null;
 	annotationTagId?: string | null;
 	description?: string | null;
-	mockedNodes?: Array<{ name: string }>;
+	mockedNodes?: Array<{ name: string; id: string }>;
 }
 
 export interface UpdateTestResponse {