From 81258297d58e037643998c8b1b19b1b864b8adc1 Mon Sep 17 00:00:00 2001
From: lyrjie <lyrjie@gmail.com>
Date: Tue, 14 Jan 2020 16:21:12 +0300
Subject: [PATCH] Fix: ability to buy the same building multiple times (#1692)

---
 core/src/com/unciv/ui/cityscreen/ConstructionsTable.kt | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/core/src/com/unciv/ui/cityscreen/ConstructionsTable.kt b/core/src/com/unciv/ui/cityscreen/ConstructionsTable.kt
index 5dcd1c02ac..e1ce998ae9 100644
--- a/core/src/com/unciv/ui/cityscreen/ConstructionsTable.kt
+++ b/core/src/com/unciv/ui/cityscreen/ConstructionsTable.kt
@@ -251,6 +251,7 @@ class ConstructionsTable(val cityScreen: CityScreen) : Table(CameraStageBaseScre
                     && !city.isPuppet) {
                 button.onClick {
                     cityConstructions.addToQueue(construction.name)
+                    if (!construction.shouldBeDisplayed(cityConstructions)) cityScreen.selectedConstruction = null
                     cityScreen.update()
                 }
             } else {
@@ -286,6 +287,7 @@ class ConstructionsTable(val cityScreen: CityScreen) : Table(CameraStageBaseScre
                         selectedQueueEntry = -2
                         cityScreen.selectedConstruction = null
                     }
+                    if (!construction.shouldBeDisplayed(cityConstructions)) cityScreen.selectedConstruction = null
                     cityScreen.update()
                 }, cityScreen)
             }