diff --git a/src/vs/workbench/contrib/notebook/browser/viewParts/notebookEditorStickyScroll.ts b/src/vs/workbench/contrib/notebook/browser/viewParts/notebookEditorStickyScroll.ts
index 0f421871f3c81..e8eb6c5a7ec13 100644
--- a/src/vs/workbench/contrib/notebook/browser/viewParts/notebookEditorStickyScroll.ts
+++ b/src/vs/workbench/contrib/notebook/browser/viewParts/notebookEditorStickyScroll.ts
@@ -89,7 +89,7 @@ export class NotebookStickyLine extends Disposable {
 	}
 
 	private toggleFoldRange(currentState: CellFoldingState) {
-		const foldingController = this.notebookEditor.getContribution('workbench.notebook.foldingController') as FoldingController;
+		const foldingController = this.notebookEditor.getContribution<FoldingController>(FoldingController.id);
 
 		const index = this.entry.index;
 		const headerLevel = this.entry.level;