From 44e9a746b76eb3aa38322e5eaf547641dba9fbda Mon Sep 17 00:00:00 2001
From: Jonathan G Rennison <j.g.rennison@gmail.com>
Date: Tue, 6 Jul 2021 19:51:32 +0100
Subject: [PATCH] Fix sorting by TE being always disabled in new train purchase
 window

---
 src/build_vehicle_gui.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/build_vehicle_gui.cpp b/src/build_vehicle_gui.cpp
index d391452461..ad502388ff 100644
--- a/src/build_vehicle_gui.cpp
+++ b/src/build_vehicle_gui.cpp
@@ -1896,7 +1896,9 @@ void DisplayLocomotiveSortDropDown(Window *w, int selected)
 {
 	uint32 hidden_mask = 0;
 	/* Disable sorting by tractive effort when the original acceleration model for trains is being used. */
-	SetBit(hidden_mask, 4); // tractive effort
+	if (_settings_game.vehicle.train_acceleration_model == AM_ORIGINAL) {
+		SetBit(hidden_mask, 4); // tractive effort
+	}
 	ShowDropDownMenu(w, _sort_listing_loco, selected, WID_BV_SORT_DROPDOWN_LOCO, 0, hidden_mask);
 }