Merge branch 'eread/fix-task-helper' into 'main'

Fix task helper

See merge request gitlab-org/gitlab-docs!1847

1 files changed, 2 insertions, 2 deletions

diff --git a/lib/task_helpers.rb b/lib/task_helpers.rb
index 54f3caf6..5614ef37 100644
--- a/lib/task_helpers.rb
+++ b/lib/task_helpers.rb
@@ -20,7 +20,7 @@ end
 def retrieve_branch(slug)
   # If CI_COMMIT_REF_NAME is not defined (run locally), set it to the default branch.
   if ENV["CI_COMMIT_REF_NAME"].nil?
-    default_branch(products['repo_url'])
+    default_branch(products[slug].fetch('repo'))
   # If we're on a gitlab-docs stable branch according to the regex, catch the
   # version and assign the product stable branches correctly.
   elsif version = ENV["CI_COMMIT_REF_NAME"].match(VERSION_FORMAT)
@@ -39,7 +39,7 @@ def retrieve_branch(slug)
   # If we're NOT on a gitlab-docs stable branch, fetch the BRANCH_* environment
   # variable, and if not assigned, set to the default branch.
   else
-    ENV.fetch("BRANCH_#{slug.upcase}", default_branch(products['repo_url']))
+    ENV.fetch("BRANCH_#{slug.upcase}", default_branch(products[slug].fetch('repo')))
   end
 end