From 6ab2c8321a006c467f95be49ee17ff120fdf31c9 Mon Sep 17 00:00:00 2001
From: DCFApixels <99481254+DCFApixels@users.noreply.github.com>
Date: Mon, 24 Mar 2025 19:30:52 +0800
Subject: [PATCH] fix

---
 src/AutoInjectSystem.cs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/AutoInjectSystem.cs b/src/AutoInjectSystem.cs
index 787ecea..ff791a2 100644
--- a/src/AutoInjectSystem.cs
+++ b/src/AutoInjectSystem.cs
@@ -101,10 +101,7 @@ namespace DCFApixels.DragonECS
         private Type[] _relatedTypesBuffer;
         public void Inject(Type fieldType, object obj)
         {
-            if (_isPreInitInjectionComplete == false)
-            {
-                _notInjected.Remove(fieldType);
-            }
+
             if (_relatedTypesBuffer == null || _relatedTypesBuffer.Length < _injectedTypeToPropertiesMap.Count)
             {
                 _relatedTypesBuffer = new Type[_injectedTypeToPropertiesMap.Count];
@@ -132,6 +129,10 @@ namespace DCFApixels.DragonECS
                         string propertyName = item.Attribute.NamedInjection;
                         if (string.IsNullOrEmpty(propertyName) || propertyName == name)
                         {
+                            if (_isPreInitInjectionComplete == false)
+                            {
+                                _notInjected.Remove(item.property.PropertyType);
+                            }
                             item.property.Inject(item.target, obj);
                         }
                     }