diff --git a/src/vs/workbench/contrib/chat/browser/codeBlockPart.ts b/src/vs/workbench/contrib/chat/browser/codeBlockPart.ts
index 5f0104bfd2e..e687eddef5a 100644
--- a/src/vs/workbench/contrib/chat/browser/codeBlockPart.ts
+++ b/src/vs/workbench/contrib/chat/browser/codeBlockPart.ts
@@ -60,6 +60,7 @@ import { isEqual } from 'vs/base/common/resources';
 import { DefaultModelSHA1Computer } from 'vs/editor/common/services/modelService';
 import { IDialogService } from 'vs/platform/dialogs/common/dialogs';
 import { TextModelText } from 'vs/editor/common/model/textModelText';
+import { MessageController } from 'vs/editor/contrib/message/browser/messageController';
 
 const $ = dom.$;
 
@@ -277,6 +278,7 @@ export class CodeBlockPart extends Disposable {
 				BracketMatchingController.ID,
 				SmartSelectController.ID,
 				HoverController.ID,
+				MessageController.ID,
 				GotoDefinitionAtPositionEditorContribution.ID,
 			])
 		}));