From 4a18d41b97e531eb1fca737e9237fa9027f69ab1 Mon Sep 17 00:00:00 2001
From: Mikhail <99481254+DCFApixels@users.noreply.github.com>
Date: Fri, 17 May 2024 00:50:05 +0800
Subject: [PATCH] fix

---
 .../Editor/ComponentTemplatePropertyDrawer.cs          | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/EntityTemplate/Editor/ComponentTemplatePropertyDrawer.cs b/src/EntityTemplate/Editor/ComponentTemplatePropertyDrawer.cs
index 3546089..64f7f3e 100644
--- a/src/EntityTemplate/Editor/ComponentTemplatePropertyDrawer.cs
+++ b/src/EntityTemplate/Editor/ComponentTemplatePropertyDrawer.cs
@@ -82,6 +82,10 @@ namespace DCFApixels.DragonECS.Unity.Editors
 
         public override float GetPropertyHeight(SerializedProperty property, GUIContent label)
         {
+            if(property.propertyType == SerializedPropertyType.ManagedReference == false)
+            {
+                return EditorGUI.GetPropertyHeight(property, label);
+            }
             IComponentTemplate template = property.managedReferenceValue as IComponentTemplate;
             if (template == null || property.managedReferenceValue == null)
             {
@@ -114,6 +118,12 @@ namespace DCFApixels.DragonECS.Unity.Editors
 
         public override void OnGUI(Rect position, SerializedProperty componentRefProp, GUIContent label)
         {
+            if (componentRefProp.propertyType == SerializedPropertyType.ManagedReference == false)
+            {
+                EditorGUI.PropertyField(position, componentRefProp, label, true);
+                return;
+            }
+
             Init();
             var counter = componentRefProp.Copy();