diff --git a/src/vs/workbench/contrib/notebook/browser/view/renderers/codeCell.ts b/src/vs/workbench/contrib/notebook/browser/view/renderers/codeCell.ts
index f22e6fea28c..74e072419b0 100644
--- a/src/vs/workbench/contrib/notebook/browser/view/renderers/codeCell.ts
+++ b/src/vs/workbench/contrib/notebook/browser/view/renderers/codeCell.ts
@@ -122,6 +122,12 @@ export class CodeCell extends Disposable {
 			}
 		}));
 
+		this._register(viewCell.onDidChangeLayout((e) => {
+			if (e.totalHeight) {
+				this.relayoutCell();
+			}
+		}));
+
 		this._register(templateData.editor.onDidContentSizeChange((e) => {
 			if (e.contentHeightChanged) {
 				if (this.viewCell.layoutInfo.editorHeight !== e.contentHeight) {