diff --git a/src/main/java/com/pfefferminzia/petinsurance/calculations/Calculations.java b/src/main/java/com/pfefferminzia/petinsurance/calculations/Calculations.java index 6645004..0854be1 100644 --- a/src/main/java/com/pfefferminzia/petinsurance/calculations/Calculations.java +++ b/src/main/java/com/pfefferminzia/petinsurance/calculations/Calculations.java @@ -23,8 +23,7 @@ public class Calculations { TariffOption tariff = tariffConfig.getTariffFactor(contract.tariffName(), insuranceStartDate); - String speciesKey = animal.getClass().getSimpleName().toUpperCase(); - BigDecimal baseInsuranceSum = tariffConfig.getAnimalBaseSum(speciesKey); + BigDecimal baseInsuranceSum = tariffConfig.getAnimalBaseSum(animal.speciesKey()); BigDecimal insuranceSum = getInsuranceSum(baseInsuranceSum, tariff.factor());