Remove all parts of big unit

Yankes · Jan 15, 2017 · 3b8032f · 3b8032f
1 parent a3313b3
commit 3b8032f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/Savegame/SavedBattleGame.cpp b/src/Savegame/SavedBattleGame.cpp
@@ -1878,13 +1878,15 @@ void SavedBattleGame::reviveUnconsciousUnits(bool noTU)
  */
 void SavedBattleGame::removeUnconsciousBodyItem(BattleUnit *bu)
 {
+	int size = bu->getArmor()->getSize();
+	size *= size;
 	// remove the unconscious body item corresponding to this unit
 	for (std::vector<BattleItem*>::iterator it = getItems()->begin(); it != getItems()->end(); )
 	{
 		if ((*it)->getUnit() == bu)
 		{
 			removeItem((*it));
-			break;
+			if (--size == 0) break;
 		}
 		++it;
 	}