diff --git a/flow/source/nvfloweditor/EditorImgui.cpp b/flow/source/nvfloweditor/EditorImgui.cpp
index 3e91b21fd..0430f4316 100644
--- a/flow/source/nvfloweditor/EditorImgui.cpp
+++ b/flow/source/nvfloweditor/EditorImgui.cpp
@@ -399,6 +399,8 @@ void editorImgui_update(
                     ImGui::TreePop();
                 }
             }
+
+            ImGui::TreePop();
         }
 
         ImGui::End();