diff --git a/scripts/sd_utils/__init__.py b/scripts/sd_utils/__init__.py
index 6f28a36c8..7f83b4f2f 100644
--- a/scripts/sd_utils/__init__.py
+++ b/scripts/sd_utils/__init__.py
@@ -16,6 +16,7 @@
 # base webui import and utils.
 # from webui_streamlit import st
 import hydralit as st
+import hydralit_components as hc
 
 # streamlit imports
 from streamlit.runtime.scriptrunner import StopException
@@ -23,7 +24,9 @@
 # from streamlit.runtime.scriptrunner import script_run_context
 
 # streamlit components section
-from streamlit_server_state import server_state, server_state_lock
+from streamlit_server_state import server_state, server_state_lock, no_rerun
+import streamlit_nested_layout
+
 
 # from streamlitextras.threader import lock, trigger_rerun, \
 # streamlit_thread, get_thread, \