diff --git a/examples/themes.js b/examples/themes.js
index 7224ff24..b2cb967c 100644
--- a/examples/themes.js
+++ b/examples/themes.js
@@ -4,7 +4,7 @@
   $(document).ready(() => {
     let body=document.querySelector('body');
     let toggleButton=document.getElementById('myonoffswitch');
-    let toggle = localStorage.getItem('currentToggle');
+    let toggle = localStorage.getItem('currentToggle') || 'day';
 
     // Sets images' theme accordingly when the page is loaded
     setTimeout(()=> {
@@ -57,10 +57,10 @@
 
       setTimeout(()=> {
         let images=document.querySelectorAll("img");
-        let check=localStorage.getItem('currentToggle');
         
-        if(check==="night") {
+        if(toggle==="night") {
           localStorage.setItem('currentToggle', 'day');
+          toggle = 'day';
           body.style.backgroundColor = '#f8f8fa';
           body.style.filter = 'invert(0)';
           body.style.transitionProperty = 'filter';
@@ -80,6 +80,7 @@
 
         }else {
           localStorage.setItem('currentToggle', 'night');
+          toggle = 'night';
           body.style.backgroundColor = '#111111';
           body.style.filter = 'invert(1)';
           body.style.transitionProperty = 'filter';
@@ -131,4 +132,4 @@
     $('.past').click((e) => buttonClickCallbackNightTheme(e));
     $('.sort-options').click((e) => buttonClickCallbackNightTheme(e));
 
-})
\ No newline at end of file
+})