diff --git a/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte b/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte
index c2919e41d7..78596ee71d 100644
--- a/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte
+++ b/frontend/src/lib/components/flows/map/FlowModuleSchemaMap.svelte
@@ -108,6 +108,7 @@
 				module.id,
 				module.summary
 			)
+			$flowStateStore[module.id] = state
 		}
 		if (!modules) return [module]
 		modules.splice(index, 0, module)