diff --git a/theme/main.html b/theme/main.html
index 3c063c6c7862d92d6097d162db0360cd038b9dec..35f8d044950b1741a0f92d2179f698cad78455a3 100644
--- a/theme/main.html
+++ b/theme/main.html
@@ -41,5 +41,9 @@
 {% endblock %}
 
 {% block content %}
+{% if 'thebelab-code' in super() %}
 {{ super() | replace('</h1>', '</h1> <button id="thebelab-activate-button" class="thebelab-button" onclick="initThebelab()">Make live</button>', count=1) }}
+{% else %}
+{{ super() }}
+{% endif %}
 {% endblock %}
\ No newline at end of file