fix possible null reference exceptions

This commit is contained in:
Matthias Mailänder
2015-05-31 15:55:28 +02:00
parent a12e6a191b
commit cd5c0eca79

View File

@@ -17,6 +17,7 @@ namespace OpenRA.Mods.Common.AI
public void Update(Squad squad) public void Update(Squad squad)
{ {
if (currentState != null)
currentState.Tick(squad); currentState.Tick(squad);
} }
@@ -31,6 +32,7 @@ namespace OpenRA.Mods.Common.AI
if (newState != null) if (newState != null)
currentState = newState; currentState = newState;
if (currentState != null)
currentState.Activate(squad); currentState.Activate(squad);
} }