diff --git a/frontend/src/lib/components/InputTransformForm.svelte b/frontend/src/lib/components/InputTransformForm.svelte
index f5aa30c546..52ea222c6a 100644
--- a/frontend/src/lib/components/InputTransformForm.svelte
+++ b/frontend/src/lib/components/InputTransformForm.svelte
@@ -47,11 +47,11 @@
 
 	let propertyType = getPropertyType(arg)
 
-	// function updatePropertyType() {
-	// 	propertyType = arg.type
-	// }
+	function updatePropertyType() {
+		propertyType = arg.type
+	}
 
-	// $: arg.type !== propertyType && updatePropertyType()
+	$: arg?.type !== propertyType && updatePropertyType()
 
 	function getPropertyType(arg: InputTransform | any): 'static' | 'javascript' {
 		let type: 'static' | 'javascript' = arg?.type ?? 'static'