Simplified update_labels() in country.

ra314 · Aug 5, 2021 · 86a3eb8 · 86a3eb8
1 parent 2b4f387
commit 86a3eb8
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/Scripts/Levels/Level Components/Country.gd b/Scripts/Levels/Level Components/Country.gd
@@ -40,10 +40,6 @@ func change_ownership_to(player):
 		get_node("Sprite").change_color_to("gray")
 
 func update_labels():
-	# Level Creator Behaviour
-	if get_tree().get_current_scene().get_name() == "Level Creator":
-		get_node("Label").text = str(num_troops)
-		return
 	get_node("Label").text = str(num_troops)
 	if belongs_to:
 		belongs_to.update_labels()
@@ -173,6 +169,7 @@ func randomise_troops():
 		num_troops = 3
 	elif rand_num < 10:
 		num_troops = 4
+	update_labels()
 
 func init(_x, _y, _country_name):
 	position = Vector2(_x, _y)