The internal fuel item determines the fuel capacity of the train. Locomotives can only have a burner-type energy source, which means it has to have something 'burning'. So the easiest way to keep the mod performant is to make the train's internal battery a 'burnable' item and manipulate its energy value when recharging.
