diff --git a/frontend/dataset_generate_page.py b/frontend/dataset_generate_page.py
index dd57cb6..cceefa4 100644
--- a/frontend/dataset_generate_page.py
+++ b/frontend/dataset_generate_page.py
@@ -81,7 +81,7 @@ def dataset_generate_page():
 
         
         doc_dropdown.change(on_doc_change, inputs=doc_dropdown, outputs=doc_state)
-        prompt_dropdown.change(on_prompt_change, inputs=prompt_dropdown, outputs=prompt_state)
+        prompt_dropdown.change(on_prompt_change, inputs=prompt_dropdown, outputs=[prompt_state, variables_dataframe])
 
         generate_button.click(
             on_generate_click,