diff --git a/pynecone/app.py b/pynecone/app.py
index df0aa9566..f1f2e3785 100644
--- a/pynecone/app.py
+++ b/pynecone/app.py
@@ -610,6 +610,9 @@ def upload(app: App):
         )
         # TODO: refactor this to handle yields.
         update = await state._process(event).__anext__()
+
+        # Set the state for the session.
+        app.state_manager.set_state(event.token, state)
         return update
 
     return upload_file