From 2c7db7e32fe3734b4d26c6ed21e8e02db56c3185 Mon Sep 17 00:00:00 2001
From: Mikhail <99481254+DCFApixels@users.noreply.github.com>
Date: Wed, 18 Sep 2024 20:05:54 +0800
Subject: [PATCH] fix reference button attribute

---
 src/Editor/ReferenceButtonAttribute.cs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Editor/ReferenceButtonAttribute.cs b/src/Editor/ReferenceButtonAttribute.cs
index 5890965..f33e9b0 100644
--- a/src/Editor/ReferenceButtonAttribute.cs
+++ b/src/Editor/ReferenceButtonAttribute.cs
@@ -38,6 +38,7 @@ namespace DCFApixels.DragonECS.Unity.Editors
                 {
                     fieldType = fieldType.GetGenericTypeDefinition();
                 }
+
                 if (fieldType == referenceBaseType)
                 {
                     _isReferenceWrapper = true;
@@ -51,11 +52,11 @@ namespace DCFApixels.DragonECS.Unity.Editors
 
         public override float GetPropertyHeight(SerializedProperty property, GUIContent label)
         {
+            Init();
             if (_isReferenceWrapper)
             {
                 property.Next(true);
             }
-
             if (property.managedReferenceValue != null)
             {
                 return EditorGUI.GetPropertyHeight(property, label, true);