diff --git a/force_bdss/core_plugins/dummy/kpi_adder/tests/test_kpi_adder_calculator.py b/force_bdss/core_plugins/dummy/kpi_adder/tests/test_kpi_adder_calculator.py
index 6f325effe44bfaa8ef23b3a16faa89c652c0b7e6..93fc276c27bfd699d3c27d2141f2437a9fbd9798 100644
--- a/force_bdss/core_plugins/dummy/kpi_adder/tests/test_kpi_adder_calculator.py
+++ b/force_bdss/core_plugins/dummy/kpi_adder/tests/test_kpi_adder_calculator.py
@@ -22,6 +22,7 @@ class TestKPIAdderCalculator(unittest.TestCase):
         model.cuba_type_out = "TOTAL_PRESSURE"
         dv1 = DataValue(type="PRESSURE", value=10)
         dv2 = DataValue(type="PRESSURE", value=30)
-        res = kpic.run(model, [dv1, dv2])
+        dv3 = DataValue(type="VOLUME", value=100)
+        res = kpic.run(model, [dv1, dv2, dv3])
         self.assertEqual(res[0].type, "TOTAL_PRESSURE")
         self.assertEqual(res[0].value, 40)