diff --git a/force_bdss/kpi/tests/test_base_kpi_calculator_model.py b/force_bdss/kpi/tests/test_base_kpi_calculator_model.py
index d382046c73a4a0f1ed5b4d695da5437ddf4bd906..c8f36d330875eb3822b18a5d28a2580673641e93 100644
--- a/force_bdss/kpi/tests/test_base_kpi_calculator_model.py
+++ b/force_bdss/kpi/tests/test_base_kpi_calculator_model.py
@@ -56,12 +56,12 @@ class TestBaseKPICalculatorModel(unittest.TestCase):
                     {
                         "__traits_version__": "4.6.0",
                         "name": "baz",
-                        "kpi": False,
+                        "is_kpi": False,
                     },
                     {
                         "__traits_version__": "4.6.0",
                         "name": "quux",
-                        "kpi": False,
+                        "is_kpi": False,
                     }
                 ]
             })