diff options
Diffstat (limited to 'src/ai/api/ai_engine.cpp')
-rw-r--r-- | src/ai/api/ai_engine.cpp | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/src/ai/api/ai_engine.cpp b/src/ai/api/ai_engine.cpp index 3948a406a..669c44ec5 100644 --- a/src/ai/api/ai_engine.cpp +++ b/src/ai/api/ai_engine.cpp @@ -6,9 +6,10 @@ #include "ai_cargo.hpp" #include "../../company_func.h" #include "../../strings_func.h" -#include "../../aircraft.h" #include "../../vehicle_func.h" #include "../../settings_type.h" +#include "../../rail.h" +#include "../../engine_base.h" #include "../../articulated_vehicles.h" #include "table/strings.h" @@ -77,11 +78,8 @@ } break; case VEH_SHIP: - return e->u.ship.capacity; - break; - case VEH_AIRCRAFT: - return AircraftDefaultCargoCapacity(e->GetDefaultCargoType(), &e->u.air); + return e->GetDisplayDefaultCapacity(); break; default: NOT_REACHED(); |