diff --git a/assets/js/custom.js b/assets/js/custom.js
index 0dfb2b8..2e53b1a 100644
--- a/assets/js/custom.js
+++ b/assets/js/custom.js
@@ -1,18 +1,12 @@
 // Put your custom JS code here
 
 // self-hosting view.
-
-const storedSelfhost = localStorage.getItem('selfhost');
-
 const getSelfhost = () => {
-  if (storedSelfhost) {
-    return storedSelfhost;
-  }
-
-  return 'no';
+  return localStorage.getItem('selfhost') || 'no';
 };
 
-const setSelfhost = function (selfhost) {
+const setSelfhost = (selfhost) => {
+  localStorage.setItem('selfhost', selfhost);
   document.documentElement.setAttribute('data-bs-selfhost', selfhost);
 };
 
@@ -22,7 +16,6 @@ window.addEventListener('DOMContentLoaded', () => {
   document.querySelectorAll('[data-bs-selfhost-value]').forEach((element) => {
     element.addEventListener('click', (event) => {
       const selfhost = element.getAttribute('data-bs-selfhost-value');
-      localStorage.setItem('selfhost', selfhost);
       setSelfhost(selfhost);
     });
   }