|
|
@@ -211,9 +211,9 @@ TEST_F(CGameStateTest, issue2765)
|
|
|
|
|
|
for(const CStack * s : gameState->curB->stacks)
|
|
|
{
|
|
|
- if(s->type->idNumber == CreatureID::BALLISTA)
|
|
|
+ if(s->type->idNumber == CreatureID::BALLISTA && s->unitSide() == BattleSide::DEFENDER)
|
|
|
def = s;
|
|
|
- else if(s->type->idNumber == CreatureID(69))
|
|
|
+ else if(s->type->idNumber == CreatureID(69) && s->unitSide() == BattleSide::ATTACKER)
|
|
|
att = s;
|
|
|
}
|
|
|
ASSERT_NE(att, nullptr);
|