diff --git a/exampleSite/config.yaml b/exampleSite/config.yaml
index 1d1fd7f..670c99c 100644
--- a/exampleSite/config.yaml
+++ b/exampleSite/config.yaml
@@ -69,6 +69,7 @@ params:
 
   # GitHub repo URL of your site
   gitRepo: https://github.com/hugo-toha/hugo-toha.github.io
+  gitBranch: main
 
   # specify whether you want to write some blog posts or not
   enableBlogPost: true
diff --git a/layouts/_default/single.html b/layouts/_default/single.html
index a83c6b3..1751c09 100644
--- a/layouts/_default/single.html
+++ b/layouts/_default/single.html
@@ -57,7 +57,7 @@
         <!--- Improve this page button --->
         {{ if site.Params.GitRepo }}
           <div class="btn-improve-page">
-              <a href="{{ site.Params.GitRepo }}/edit/master/content/{{ .File.Path }}">
+              <a href="{{ site.Params.GitRepo }}/edit/{{ site.Params.GitBranch }}/content/{{ .File.Path }}">
                 <i class="fas fa-code-branch"></i>
                 {{ i18n "improve_this_page" }}
               </a>