BaseSubStateMachineState - Added a call to ReplaceModelWithActiveModel after creating the state machine in SwitchModel

Author: PaulNonatomic

Runtime/StateGraph/States/BaseSubStateMachineState.cs

@@ -85,6 +85,7 @@ protected virtual void SwitchModel(StateMachineModel value)
 
 			_model = value;
 			CreateStateMachine();
+			ReplaceModelWithActiveModel();
 
 			if (!GameObject.activeInHierarchy || !_started) return;
 			SubStateMachine?.Start();
@@ -106,14 +107,14 @@ protected virtual void OnSubStateComplete(State state, StateMachineModel model)
 			//...
 		}
 
-		private void ReplaceModelWithActiveModel()
+		protected virtual void ReplaceModelWithActiveModel()
 		{
 			if (SubStateMachine == null) return;
 
 			_model = SubStateMachine.Model;
 		}
 
-		private void ReplaceModelWithOriginalModel()
+		protected virtual void ReplaceModelWithOriginalModel()
 		{
 			if (!_model) return;