diff --git a/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte b/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
index d16623cf8d..8d12d51aa8 100644
--- a/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
+++ b/frontend/src/lib/components/flows/content/FlowModuleComponent.svelte
@@ -124,14 +124,18 @@
 		try {
 			const { input_transforms, schema } = await loadSchemaFromModule(flowModule)
 			validCode = true
-			inputTransformSchemaForm?.setArgs(input_transforms)
-			if (
-				flowModule.value.type == 'rawscript' ||
-				flowModule.value.type == 'script' ||
-				flowModule.value.type == 'flow'
-			) {
-				flowModule.value.input_transforms = input_transforms
+			if (inputTransformSchemaForm) {
+				inputTransformSchemaForm.setArgs(input_transforms)
+			} else {
+				if (
+					flowModule.value.type == 'rawscript' ||
+					flowModule.value.type == 'script' ||
+					flowModule.value.type == 'flow'
+				) {
+					flowModule.value.input_transforms = input_transforms
+				}
 			}
+
 			if (flowModule.value.type == 'rawscript' && flowModule.value.lock != undefined) {
 				flowModule.value.lock = undefined
 			}