diff --git a/public/site/controllers/site.php b/public/site/controllers/site.php
index 07de5ec..c7bf096 100644
--- a/public/site/controllers/site.php
+++ b/public/site/controllers/site.php
@@ -1,6 +1,9 @@
 <?php
 
-return function ($page, $kirby) {
+return function ($page, $kirby, $site) {
+  if (!$kirby->user()) {
+    go($site->panel()->url());
+  }
   $data = $page->toArray();
   $data['template'] = (string) $page->template();
 
diff --git a/public/site/snippets/generic-template.php b/public/site/snippets/generic-template.php
index e5d4929..423037d 100644
--- a/public/site/snippets/generic-template.php
+++ b/public/site/snippets/generic-template.php
@@ -1,9 +1,3 @@
 <?php snippet('header') ?>
-<?php 
-  if ($kirby->user()) {
-    echo '<div id="app"></div>';
-  } else {
-    go($site->panel()->url());
-  }
-?>
+<div id="app"></div>
 <?php snippet('footer') ?>
\ No newline at end of file