From 6f5edbae0cdf97b8717b805619b9147e9c08965b Mon Sep 17 00:00:00 2001
From: Yair Morgenstern <yairm210@hotmail.com>
Date: Tue, 3 Jan 2023 23:04:03 +0200
Subject: [PATCH] Resolve #4811 - Allow mass unit upgrades from the unit
 overview screen

---
 core/src/com/unciv/ui/overviewscreen/UnitOverviewTable.kt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/core/src/com/unciv/ui/overviewscreen/UnitOverviewTable.kt b/core/src/com/unciv/ui/overviewscreen/UnitOverviewTable.kt
index b0b73d38c1..7a098c3a01 100644
--- a/core/src/com/unciv/ui/overviewscreen/UnitOverviewTable.kt
+++ b/core/src/com/unciv/ui/overviewscreen/UnitOverviewTable.kt
@@ -151,6 +151,7 @@ class UnitOverviewTab(
     }
 
     private fun Table.updateUnitListTable(): Table {
+        clear()
         val game = overviewScreen.game
         defaults().pad(5f)
 
@@ -225,9 +226,9 @@ class UnitOverviewTab(
                 val upgradeIcon = ImageGetter.getUnitIcon(unit.getUnitToUpgradeTo().name,
                     if (enable) Color.GREEN else Color.GREEN.darken(0.5f))
                 if (enable) upgradeIcon.onClick {
-                    showWorldScreenAt(unit)
                     SoundPlayer.play(unitAction!!.uncivSound)
                     unitAction.action!!()
+                    unitListTable.updateUnitListTable()
                 }
                 add(upgradeIcon).size(28f)
             } else add()