diff options
Diffstat (limited to 'src/vehicle.cpp')
-rw-r--r-- | src/vehicle.cpp | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/src/vehicle.cpp b/src/vehicle.cpp index d3b47c7e3..cb915a6f7 100644 --- a/src/vehicle.cpp +++ b/src/vehicle.cpp @@ -1665,10 +1665,8 @@ PaletteID GetEnginePalette(EngineID engine_type, CompanyID company) PaletteID GetVehiclePalette(const Vehicle *v) { - if (v->type == VEH_TRAIN) { - return GetEngineColourMap(v->engine_type, v->owner, Train::From(v)->tcache.first_engine, v); - } else if (v->type == VEH_ROAD) { - return GetEngineColourMap(v->engine_type, v->owner, RoadVehicle::From(v)->rcache.first_engine, v); + if (v->IsGroundVehicle()) { + return GetEngineColourMap(v->engine_type, v->owner, v->GetGroundVehicleCache()->first_engine, v); } return GetEngineColourMap(v->engine_type, v->owner, INVALID_ENGINE, v); |