diff --git a/knowledge-space.php b/knowledge-space.php
index 28e0b4c3f3aebc944470b1531423927ada1a8483..783d2afe7f686add29688c5e0ab3899ab41ca503 100644
--- a/knowledge-space.php
+++ b/knowledge-space.php
@@ -15,7 +15,7 @@ function ks_add_graph($atts = []): string
 
     $script_path = 'build' . DIRECTORY_SEPARATOR . $GLOBALS['build'] . DIRECTORY_SEPARATOR . 'graph.js';
     wp_enqueue_script("ks-display-js", plugins_url($script_path, __FILE__), array('jquery'), false);
-    ks_localize("ks-display-js");
+    ks_localize("ks-display-js", $atts);
 
     return $div;
 }
@@ -25,7 +25,7 @@ function ks_echo_graph($atts = []): void
     echo ks_add_graph($atts);
 }
 
-function ks_localize($handle)
+function ks_localize($handle, $atts)
 {
     $space_id = kg_get_space_id_from_atts($atts);
     $plugin_dir = plugin_dir_url(__FILE__);