diff --git a/assets/runtime/functions b/assets/runtime/functions
index ffb066e7..98490422 100644
--- a/assets/runtime/functions
+++ b/assets/runtime/functions
@@ -281,9 +281,9 @@ gitlab_configure_gitaly() {
     GITALY_SOCKET_PATH \
     GITLAB_REPOS_DIR
 
-  if [[ ${GITALY_ENABLED} != true ]]; then
-    exec_as_git sed -i "/{{GITALY_ADDRESS}}/d" ${GITLAB_CONFIG}
-  fi
+  update_template ${GITLAB_CONFIG} \
+    GITALY_ENABLED
+  
 }
 
 gitlab_configure_gitlab_workhorse() {