diff --git a/src/test/java/com/schematic/api/datastream/EventQuantitiesTest.java b/src/test/java/com/schematic/api/datastream/EventQuantitiesTest.java
index 58d81618..b990f969 100644
--- a/src/test/java/com/schematic/api/datastream/EventQuantitiesTest.java
+++ b/src/test/java/com/schematic/api/datastream/EventQuantitiesTest.java
@@ -220,10 +220,7 @@ void aCompanyEntitlementsQuantityRatesReachTheResult() throws Exception {
                 engine.checkFlag(inferenceFlag(0.5, null), MAPPER.treeToValue(company, RulesengineCompany.class), null);
 
         assertTrue(result.getEntitlement().isPresent());
-        assertEquals(
-                MAPPER.valueToTree(RATES),
-                MAPPER.valueToTree(
-                        result.getEntitlement().get().getAdditionalProperties().get("quantity_rates")));
+        assertEquals(Optional.of(RATES), result.getEntitlement().get().getQuantityRates());
     }
 
     /**