From 2aefeec2259a6b2a146021a0b8609f88acce84e1 Mon Sep 17 00:00:00 2001
From: Jonathan G Rennison <j.g.rennison@gmail.com>
Date: Wed, 5 Jun 2024 19:44:04 +0100
Subject: [PATCH] Fix infrastructure totals when removing canals in
 ForceClearWaterTile

---
 src/water_cmd.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/water_cmd.cpp b/src/water_cmd.cpp
index 98436b66aa..da6181bdb8 100644
--- a/src/water_cmd.cpp
+++ b/src/water_cmd.cpp
@@ -670,6 +670,13 @@ static CommandCost ClearTile_Water(TileIndex tile, DoCommandFlag flags)
 
 void ForceClearWaterTile(TileIndex tile)
 {
+	if (IsWaterTile(tile) && IsCanal(tile)) {
+		Owner owner = GetTileOwner(tile);
+		if (Company::IsValidID(owner)) {
+			Company::Get(owner)->infrastructure.water--;
+			DirtyCompanyInfrastructureWindows(owner);
+		}
+	}
 	bool remove = IsDockingTile(tile);
 	DoClearSquare(tile);
 	MarkCanalsAndRiversAroundDirty(tile);