diff --git a/layouts/partials/header.html b/layouts/partials/header.html
index 33c7b42..c77f7cf 100644
--- a/layouts/partials/header.html
+++ b/layouts/partials/header.html
@@ -26,7 +26,7 @@
   {{ $is_in_is_in = $is_in_is_in | append ($.GetPage .) }}
 {{ end }}
 
-{{ if eq hugo.Environment "preview" }}
+{{ if eq hugo.Environment "gitea" }}
   <div class="headiter">
     <a class="edit-button" target="_blank" href="{{ $.Site.Params.sandpointsGiteaUrl }}/_edit/master/content/{{ .File }}">EDIT_THIS</a>
     {{ with $scratch.core }}<a class="edit-button" target="_blank" href="{{ $.Site.Params.sandpointsGiteaUrl }}/_new/master/content/{{ index . 0 }}/">ADD_{{ index . 0 | upper }}</a>{{ end }}