diff --git a/pr_agent/settings/configuration.toml b/pr_agent/settings/configuration.toml
index 1172b169..0e9b6a3a 100644
--- a/pr_agent/settings/configuration.toml
+++ b/pr_agent/settings/configuration.toml
@@ -50,6 +50,7 @@ final_clip_factor = 0.9
 [pr_add_docs_prompt] # /add_docs #
 extra_instructions = ""
 docs_style = "Google Style" # "Google Style", "Numpy Style", "Sphinx Style", "PEP257", "reStructuredText"
+max_number_of_calls = 5
 
 [pr_update_changelog] # /update_changelog #
 push_changelog_changes=false
diff --git a/pr_agent/tools/pr_add_docs.py b/pr_agent/tools/pr_add_docs.py
index 6bf35531..d40aaf79 100644
--- a/pr_agent/tools/pr_add_docs.py
+++ b/pr_agent/tools/pr_add_docs.py
@@ -67,7 +67,7 @@ class PRAddDocs:
                 logging.info('Pushing PR review...')
                 self.git_provider.remove_initial_comment()
                 logging.info('Pushing inline code documentation...')
-                self.push_inline_docs_suggestions(data)
+                self.push_inline_docs(data)
         except Exception as e:
             logging.error(f"Failed to generate code documentation for PR, error: {e}")
 
@@ -103,8 +103,8 @@ class PRAddDocs:
             data = {'Code Documentation': data}
         return data
 
-    def push_inline_docs_suggestions(self, data):
-        code_suggestions = []
+    def push_inline_docs(self, data):
+        docs = []
 
         if not data['Code Documentation']:
             return self.git_provider.publish_comment('No code documentation found to improve this PR.')
@@ -122,18 +122,18 @@ class PRAddDocs:
                                                         add_original_line=True)
 
                     body = f"**Suggestion:** Proposed documentation\n```suggestion\n" + new_code_snippet + "\n```"
-                    code_suggestions.append({'body': body, 'relevant_file': relevant_file,
+                    docs.append({'body': body, 'relevant_file': relevant_file,
                                              'relevant_lines_start': relevant_line,
                                              'relevant_lines_end': relevant_line})
             except Exception:
                 if get_settings().config.verbosity_level >= 2:
                     logging.info(f"Could not parse code docs: {d}")
 
-        is_successful = self.git_provider.publish_code_suggestions(code_suggestions)
+        is_successful = self.git_provider.publish_code_suggestions(docs)
         if not is_successful:
             logging.info("Failed to publish code docs, trying to publish each docs separately")
-            for code_suggestion in code_suggestions:
-                self.git_provider.publish_code_suggestions([code_suggestion])
+            for documentation in docs:
+                self.git_provider.publish_code_suggestions([documentation])
 
     def dedent_code(self, relevant_file, relevant_lines_start, new_code_snippet, doc_placement='after',
                     add_original_line=False):
@@ -170,7 +170,7 @@ class PRAddDocs:
     async def _prepare_prediction_extended(self, model: str) -> dict:
         logging.info('Getting PR diff...')
         patches_diff_list = get_pr_multi_diffs(self.git_provider, self.token_handler, model,
-                                               max_calls=get_settings().pr_code_suggestions.max_number_of_calls)
+                                               max_calls=get_settings().pr_add_docs_prompt.max_number_of_calls)
 
         logging.info('Getting multi AI predictions...')
         prediction_list = []