diff --git a/apps/sim/blocks/blocks/knowledge.ts b/apps/sim/blocks/blocks/knowledge.ts
index f488c49db7..34b449373a 100644
--- a/apps/sim/blocks/blocks/knowledge.ts
+++ b/apps/sim/blocks/blocks/knowledge.ts
@@ -127,6 +127,8 @@ export const KnowledgeBlock: BlockConfig = {
       title: 'Document',
       type: 'document-selector',
       canonicalParamId: 'documentId',
+      serviceId: 'knowledge',
+      selectorKey: 'knowledge.documents',
       placeholder: 'Select document',
       dependsOn: ['knowledgeBaseSelector'],
       required: true,