summaryrefslogtreecommitdiff
path: root/player_gui.c
diff options
context:
space:
mode:
Diffstat (limited to 'player_gui.c')
-rw-r--r--player_gui.c3
1 files changed, 2 insertions, 1 deletions
diff --git a/player_gui.c b/player_gui.c
index 3d97a2612..67fa40041 100644
--- a/player_gui.c
+++ b/player_gui.c
@@ -15,6 +15,7 @@
#include "economy.h"
#include "network.h"
#include "variables.h"
+#include "train.h"
#ifdef ENABLE_NETWORK
#include "network_data.h"
@@ -432,7 +433,7 @@ static void DrawPlayerVehiclesAmount(PlayerID player)
FOR_ALL_VEHICLES(v) {
if (v->owner == player) {
switch (v->type) {
- case VEH_Train: if (v->subtype == TS_Front_Engine) train++; break;
+ case VEH_Train: if (IsFrontEngine(v)) train++; break;
case VEH_Road: road++; break;
case VEH_Aircraft: if (v->subtype <= 2) air++; break;
case VEH_Ship: ship++; break;