diff --git a/packages/syft/src/syft/util/notebook_ui/notebook_addons.py b/packages/syft/src/syft/util/notebook_ui/notebook_addons.py
index 7bb952b1d84..7d918885ef4 100644
--- a/packages/syft/src/syft/util/notebook_ui/notebook_addons.py
+++ b/packages/syft/src/syft/util/notebook_ui/notebook_addons.py
@@ -9,18 +9,20 @@
 
 CSS_CODE = """
 <style>
-    :root {
-        --primary-color: #ffffff;
-        --secondary-color: #f5f5f5;
-        --tertiary-color: #000000de;
-        --button-color: #d1d5db;
-    }
-    .dark-theme {
+  body[data-jp-theme-light='false'] {
         --primary-color: #111111;
         --secondary-color: #212121;
         --tertiary-color: #CFCDD6;
         --button-color: #111111;
-    }
+  }
+
+  body {
+        --primary-color: #ffffff;
+        --secondary-color: #f5f5f5;
+        --tertiary-color: #000000de;
+        --button-color: #d1d5db;
+  }
+
   .header-1 {
         font-style: normal;
         font-weight: 600;
@@ -561,14 +563,6 @@
                         buildPaginationContainer${uid}(result);
                     }
 
-                    function setTheme${uid}(){
-                        const theme = document.body.getAttribute('data-jp-theme-light') === 'true' ? 'light' : 'dark';
-
-                        if (theme === 'dark') document.body.classList.add('dark-theme');
-                        else document.body.classList.remove('dark-theme');
-
-                    }
-
                     function resetById${uid}(id){
                         let element = document.getElementById(id);
                         while (element.firstChild) {
@@ -583,7 +577,6 @@
                                 let div = document.createElement("div");
                                 div.classList.add('grid-header', 'grid-index-cells');
                                 grid.appendChild(div);
-                                setTheme${uid}();
                                 headers.forEach((title) =>{
                                     let div = document.createElement("div");
                                     div.classList.add('grid-header', 'grid-std-cells');