diff --git a/test/unit/api/test_assumption_builder.py b/test/unit/api/test_assumption_builder.py
index 24f0268f4d..5d927aa90d 100644
--- a/test/unit/api/test_assumption_builder.py
+++ b/test/unit/api/test_assumption_builder.py
@@ -104,7 +104,7 @@ def test_assumptions_builder_unsuitable_available_operations():
 
     train_input, _, _ = get_dataset(task_type='classification')
     train_input = DataPreprocessor().obligatory_prepare_for_fit(train_input)
-    available_operations = ['linear', 'lagged'] # 'xgboost'
+    available_operations = ['linear', 'lagged']  # 'xgboost'
 
     default_builder = UniModalAssumptionsBuilder(train_input)
     checked_builder = UniModalAssumptionsBuilder(train_input) \