diff --git a/force_bdss/core_plugins/test_kpi/kpi_adder/kpi_adder_calculator.py b/force_bdss/core_plugins/test_kpi/kpi_adder/kpi_adder_calculator.py
index c7554ff706b4557df8aea48ca29646a322bd9d19..3bb01b9b43cd74607e2f61994526988c9dc80670 100644
--- a/force_bdss/core_plugins/test_kpi/kpi_adder/kpi_adder_calculator.py
+++ b/force_bdss/core_plugins/test_kpi/kpi_adder/kpi_adder_calculator.py
@@ -17,7 +17,6 @@ class KPIAdderCalculator(BaseKPICalculator):
 
         return KPICalculatorResult(
             originator=self,
-            value_types = [self.model.cuba_type_out],
+            value_types=[self.model.cuba_type_out],
             values=numpy.array([sum])
         )
-
diff --git a/force_bdss/core_plugins/test_kpi/kpi_adder/kpi_adder_model.py b/force_bdss/core_plugins/test_kpi/kpi_adder/kpi_adder_model.py
index 066035c3f74ef0deb99eb113c486e5250fe02fe8..159f873cf4f255d07a708d68dfeb9ca55b2dafcf 100644
--- a/force_bdss/core_plugins/test_kpi/kpi_adder/kpi_adder_model.py
+++ b/force_bdss/core_plugins/test_kpi/kpi_adder/kpi_adder_model.py
@@ -1,4 +1,4 @@
-from traits.api import HasStrictTraits, Int, String
+from traits.api import HasStrictTraits, String
 
 
 class KPIAdderModel(HasStrictTraits):