diff --git a/ote_sdk/ote_sdk/test_suite/training_tests_actions.py b/ote_sdk/ote_sdk/test_suite/training_tests_actions.py
index 8e5fc0b380b..76e276c7395 100644
--- a/ote_sdk/ote_sdk/test_suite/training_tests_actions.py
+++ b/ote_sdk/ote_sdk/test_suite/training_tests_actions.py
@@ -79,7 +79,7 @@ def __call__(self, data_collector: DataCollector, results_prev_stages: OrderedDi
 
 
 def create_environment_and_task(
-    params, labels_schema, model_template, dataset, model_adapters=None
+    params, labels_schema, model_template, dataset=None, model_adapters=None
 ):
 
     environment = TaskEnvironment(