Fix UnitAI states in PetraAI.
Sometimes PetraAI relies on UnitAI state of an entity, but those were not changed after modifying UnitAI.
Introduced in r24989 / rP24989 and r25345 / rP25345.
Differential revision: https://code.wildfiregames.com/D3976