diff --git a/streamlit_server_state/_compat.py b/streamlit_server_state/_compat.py
index 2bebdd7..00fd09f 100644
--- a/streamlit_server_state/_compat.py
+++ b/streamlit_server_state/_compat.py
@@ -40,9 +40,8 @@
 Ref: https://github.com/streamlit/streamlit/pull/5856
 """
 
-
 try:
-    from streamlit.runtime.scriptrunner.script_run_context import (
+    from streamlit.runtime.scriptrunner_utils.script_run_context import (
         SCRIPT_RUN_CONTEXT_ATTR_NAME,
         ScriptRunContext,
     )