diff --git a/adala/skills/collection/prompt_improvement.py b/adala/skills/collection/prompt_improvement.py
index ce8c555c..30c86fdf 100644
--- a/adala/skills/collection/prompt_improvement.py
+++ b/adala/skills/collection/prompt_improvement.py
@@ -70,6 +70,9 @@ def get_json_template(fields):
         else:
             model_json_schema = self.skill_to_improve.response_model.model_json_schema()
 
+        # TODO: can remove this when only LabelStudioSkill is supported
+        label_config = getattr(self.skill_to_improve, 'label_config', '<View>Not available</View>')
+
         input_variables = self.input_variables
         output_variables = list(model_json_schema['properties'].keys())
         input_json_template = get_json_template(input_variables)
@@ -84,7 +87,7 @@ def get_json_template(fields):
 
 ## Current Labeling Config:
 ```xml
-{self.skill_to_improve.label_config}
+{label_config}
 ```
 
 ## Input variables: