diff --git a/Engine.Core/BehaviourCollector.cs b/Engine.Core/BehaviourCollector.cs index 19d8782..4eedb67 100644 --- a/Engine.Core/BehaviourCollector.cs +++ b/Engine.Core/BehaviourCollector.cs @@ -76,7 +76,7 @@ public class BehaviourCollector : IBehaviourCollector where T : class OnUniverseObjectRegistered(universe, new(universeObject)); universe.OnUniverseObjectRegistered.AddListener(delegateOnUniverseObjectRegistered); - universe.OnUniverseObjectUnRegistered.AddListener(delegateOnUniverseObjectUnregistered); + universe.OnPreUniverseObjectUnRegistered.AddListener(delegateOnUniverseObjectUnregistered); Universe = universe; OnAssign(universe); @@ -94,7 +94,7 @@ public class BehaviourCollector : IBehaviourCollector where T : class OnUniverseObjectUnregistered(Universe, new(universeObject)); Universe.OnUniverseObjectRegistered.RemoveListener(delegateOnUniverseObjectRegistered); - Universe.OnUniverseObjectUnRegistered.RemoveListener(delegateOnUniverseObjectUnregistered); + Universe.OnPreUniverseObjectUnRegistered.RemoveListener(delegateOnUniverseObjectUnregistered); Universe = null!; OnUnassigned?.Invoke(this);