Move check

packages/toc/src/generators/notebookgenerator/itemrenderer.tsx

@@ -88,6 +88,10 @@ function setCollapsedState(
   cell: Cell,
   state: boolean
 ): void {
+  // Ensure a widget is currently focused...
+  if (tracker.currentWidget === null) {
+    return;
+  }
   // Guard against attempting to collapse already hidden cells...
   if (state) {
     if (cell.isHidden) {
@@ -111,10 +115,6 @@ function setCollapsedState(
   if (level === -1) {
     return;
   }
-  // Ensure a widget is currently focused...
-  if (tracker.currentWidget === null) {
-    return;
-  }
   const widgets = tracker.currentWidget.content.widgets;
   const len = widgets.length;
   const idx = widgets.indexOf(cell);