diff --git a/Source/Editor/Surface/ContextMenu/VisjectCM.cs b/Source/Editor/Surface/ContextMenu/VisjectCM.cs
index 24cb29cb9..3a3dfe027 100644
--- a/Source/Editor/Surface/ContextMenu/VisjectCM.cs
+++ b/Source/Editor/Surface/ContextMenu/VisjectCM.cs
@@ -506,15 +506,14 @@ namespace FlaxEditor.Surface.ContextMenu
             {
                 _groups[i].ResetView();
             }
-            UnlockChildrenRecursive();
-
-            SortGroups();
-            PerformLayout();
-
             for (int i = 0; i < _groups.Count; i++)
             {
                 _groups[i].EvaluateVisibilityWithBox(_selectedBox);
             }
+            UnlockChildrenRecursive();
+
+            SortGroups();
+            PerformLayout();
             
             Profiler.EndEvent();
         }