fix possible null reference exceptions

2015-05-31 15:55:28 +02:00
parent a12e6a191b
commit cd5c0eca79
1 changed files with 4 additions and 2 deletions
--- a/OpenRA.Mods.Common/AI/StateMachine.cs
+++ b/OpenRA.Mods.Common/AI/StateMachine.cs
@@ -17,6 +17,7 @@ namespace OpenRA.Mods.Common.AI

 		public void Update(Squad squad)
 		{
+			if (currentState != null)
 				currentState.Tick(squad);
 		}

@@ -31,6 +32,7 @@ namespace OpenRA.Mods.Common.AI
 			if (newState != null)
 				currentState = newState;

+			if (currentState != null)
 				currentState.Activate(squad);
 		}