diff --git a/pr_agent/algo/utils.py b/pr_agent/algo/utils.py
index 8cf6f01a..69e52324 100644
--- a/pr_agent/algo/utils.py
+++ b/pr_agent/algo/utils.py
@@ -565,7 +565,7 @@ def try_fix_yaml(response_text: str,
     except:
         pass
 
-    # forth fallback - try to extract yaml snippet by first_key and 'last_key'
+    # forth fallback - try to extract yaml snippet by 'first_key' and 'last_key'
     # note that 'last_key' can be in practice a key that is not the last key in the yaml snippet.
     # it just needs to be some inner key, so we can look for newlines after it
     if first_key and last_key:
diff --git a/pr_agent/tools/pr_code_suggestions.py b/pr_agent/tools/pr_code_suggestions.py
index 0e39a6cf..2b409cb7 100644
--- a/pr_agent/tools/pr_code_suggestions.py
+++ b/pr_agent/tools/pr_code_suggestions.py
@@ -231,7 +231,8 @@ class PRCodeSuggestions:
 
     def _prepare_pr_code_suggestions(self, predictions: str) -> Dict:
         data = load_yaml(predictions.strip(),
-                         keys_fix_yaml=["relevant_file", "suggestion_content", "existing_code", "improved_code"])
+                         keys_fix_yaml=["relevant_file", "suggestion_content", "existing_code", "improved_code"],
+                         first_key="code_suggestions",last_key="label")
         if isinstance(data, list):
             data = {'code_suggestions': data}