diff --git a/src/vs/workbench/contrib/notebook/browser/view/cellParts/chat/cellChatController.ts b/src/vs/workbench/contrib/notebook/browser/view/cellParts/chat/cellChatController.ts
index f2f620ce9d9ba..1406eab2b188b 100644
--- a/src/vs/workbench/contrib/notebook/browser/view/cellParts/chat/cellChatController.ts
+++ b/src/vs/workbench/contrib/notebook/browser/view/cellParts/chat/cellChatController.ts
@@ -155,6 +155,8 @@ export class NotebookCellChatController extends Disposable {
 			// might not be attached
 		}
 
+		// dismiss since we can't restore  the widget properly now
+		this.dismiss(false);
 		this._widget?.dispose();
 		this._inlineChatListener?.dispose();
 		this._toolbar?.dispose();