diff --git a/docs/layouts/shortcodes/training_link.html b/docs/layouts/shortcodes/training_link.html
index 2f0e3cb5d8a72..2ff1d32d64df8 100644
--- a/docs/layouts/shortcodes/training_link.html
+++ b/docs/layouts/shortcodes/training_link.html
@@ -23,6 +23,6 @@
         - file: The absolute path to the image file (required)
         - name: The rendered link name (required)
 */}}
-<a href="{{ .Site.Params.TrainingExercises }}/blob/{{ .Site.Params.Branch }}/{{ .Get "file" }}">
+<a href="{{ .Site.Params.TrainingExercises }}/tree/{{ .Site.Params.TrainingBranch | default "master" }}/{{ .Get "file" }}">
     {{ .Get "name" }}
 </a>
diff --git a/docs/layouts/shortcodes/training_repo.html b/docs/layouts/shortcodes/training_repo.html
index 939f334facc54..fe84d8926a7dd 100644
--- a/docs/layouts/shortcodes/training_repo.html
+++ b/docs/layouts/shortcodes/training_repo.html
@@ -23,7 +23,7 @@
         - file: The absolute path to the image file (required)
         - name: The rendered link name (required)
 */}}
-<a href="{{ .Site.Params.TrainingExercises }}/tree/{{ .Site.Params.Branch }}/">
+<a href="{{ .Site.Params.TrainingExercises }}/tree/{{ .Site.Params.TrainingBranch | default "master" }}/">
     flink-training-repo
 </a>