diff --git a/invenio_theme_tugraz/assets/semantic-ui/js/invenio_theme_tugraz/theme.js b/invenio_theme_tugraz/assets/semantic-ui/js/invenio_theme_tugraz/theme.js
index 4068a46..55ba290 100644
--- a/invenio_theme_tugraz/assets/semantic-ui/js/invenio_theme_tugraz/theme.js
+++ b/invenio_theme_tugraz/assets/semantic-ui/js/invenio_theme_tugraz/theme.js
@@ -1,11 +1,8 @@
-export function hide() {
-    var x = document.getElementById('heading');
+export function toggleVisibility(id) {
+    var element = document.getElementById(id);
+    var isHided = element.style.display === "none";
 
-    if (x.style.display === "none") {
-        x.style.display = "block";
-    } else {
-        x.style.display = "none";
-    }
+    element.style.display = isHided ? "block" : "none";
 }
 
-window.hide = hide;
+window.toggleVisibility = toggleVisibility;
diff --git a/invenio_theme_tugraz/templates/invenio_theme_tugraz/header.html b/invenio_theme_tugraz/templates/invenio_theme_tugraz/header.html
index 9b5d632..ce8fa5f 100644
--- a/invenio_theme_tugraz/templates/invenio_theme_tugraz/header.html
+++ b/invenio_theme_tugraz/templates/invenio_theme_tugraz/header.html
@@ -125,7 +125,7 @@
           </div>
 
           <div class="short-menu-right">
-            <div class="short-menu-right-h" onclick="hide();">
+            <div class="short-menu-right-h" onclick="toggleVisibility('heading');">
 
               <i class="fa fa-bars fa-lg"></i><span style="margin-left:3px;">Menu</span>