From b4994b2eb74a8627ed88ff413391956034381d20 Mon Sep 17 00:00:00 2001
From: uurha <uurha.dev@gmail.com>
Date: Sun, 11 Aug 2024 04:45:19 +0200
Subject: [PATCH] Update label select query

---
 .../Editor/Drawers/Container/ElementsContainer.cs     | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/Assets/BetterCommons/Editor/Drawers/Container/ElementsContainer.cs b/Assets/BetterCommons/Editor/Drawers/Container/ElementsContainer.cs
index 5ff8fd5..cdcac6f 100644
--- a/Assets/BetterCommons/Editor/Drawers/Container/ElementsContainer.cs
+++ b/Assets/BetterCommons/Editor/Drawers/Container/ElementsContainer.cs
@@ -113,17 +113,12 @@ private void ScheduleUpdateGeometry(GeometryChangedEvent changedEvent)
         {
             if (changedEvent.target is not PropertyField element) return;
 
-            element.schedule.Execute(() => OnGeometryChanged(element));
+            element.schedule.Execute(OnGeometryChanged);
         }
 
-        private void OnGeometryChanged(PropertyField propertyField)
+        private void OnGeometryChanged()
         {
-            var labels = propertyField.Query<Label>().Class(PropertyField.labelUssClassName).Where(field =>
-            {
-                var firstAncestor = field.GetFirstAncestorOfType<PropertyField>();
-                return firstAncestor != null && firstAncestor.Equals(propertyField);
-            });
-            var label = labels.First();
+            var label = RootElement.Query<Label>().Class(PropertyField.labelUssClassName).First();
             if (label != null)
             {
                 LabelContainer.Setup(label);