From 2e629b752441732a66877b62355853a1b6dc6da9 Mon Sep 17 00:00:00 2001
From: Patrik Olesen <patrik@hemma.org>
Date: Sun, 28 Jan 2024 18:13:01 +0100
Subject: [PATCH] [18Uruguay] Update Discard train. Only trains shall end up in
 the bankpool

---
 lib/engine/game/g_18_uruguay/game.rb           |  2 +-
 .../game/g_18_uruguay/step/discard_train.rb    | 18 ++++++++++++++++++
 2 files changed, 19 insertions(+), 1 deletion(-)
 create mode 100644 lib/engine/game/g_18_uruguay/step/discard_train.rb

diff --git a/lib/engine/game/g_18_uruguay/game.rb b/lib/engine/game/g_18_uruguay/game.rb
index 340a3f7b0d..da4ac5e332 100644
--- a/lib/engine/game/g_18_uruguay/game.rb
+++ b/lib/engine/game/g_18_uruguay/game.rb
@@ -264,7 +264,7 @@ def operating_round(round_num)
             Engine::Step::Token,
             Engine::Step::Route,
             Engine::Step::Dividend,
-            Engine::Step::DiscardTrain,
+            G18Uruguay::Step::DiscardTrain,
             Engine::Step::BuyTrain,
           ], round_num: round_num)
         end
diff --git a/lib/engine/game/g_18_uruguay/step/discard_train.rb b/lib/engine/game/g_18_uruguay/step/discard_train.rb
new file mode 100644
index 0000000000..3efa97ec08
--- /dev/null
+++ b/lib/engine/game/g_18_uruguay/step/discard_train.rb
@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+require_relative '../../../step/discard_train'
+
+module Engine
+  module Game
+    module G18Uruguay
+      module Step
+        class DiscardTrain < Engine::Step::DiscardTrain
+          def process_discard_train(action)
+            action.train.remove_variants!
+            super
+          end
+        end
+      end
+    end
+  end
+end