diff --git a/daras_ai_v2/base.py b/daras_ai_v2/base.py
index 7de7c9dcd..e95ce8730 100644
--- a/daras_ai_v2/base.py
+++ b/daras_ai_v2/base.py
@@ -2377,12 +2377,12 @@ def run_as_api_tab(self):
         as_form_data = gui.checkbox("##### Upload Files via Form Data")
 
         api_url, request_body = self.get_example_request(
-            gui.session_state,
+            self.current_sr.state,
             include_all=include_all,
             pr=self.current_pr,
         )
         response_body = self.get_example_response_body(
-            gui.session_state, as_async=as_async, include_all=include_all
+            self.current_sr.state, as_async=as_async, include_all=include_all
         )
 
         api_example_generator(