diff --git a/src/frontend/src/utils/reactflowUtils.ts b/src/frontend/src/utils/reactflowUtils.ts
index 369ab9d22..943292884 100644
--- a/src/frontend/src/utils/reactflowUtils.ts
+++ b/src/frontend/src/utils/reactflowUtils.ts
@@ -640,7 +640,9 @@ function updateGroupNodeTemplate(template: APITemplateType) {
         type === "code" ||
         type === "prompt" ||
         type === "file" ||
-        type === "int") &&
+        type === "int" ||
+        type === "dict" ||
+        type === "NestedDict") &&
       !template[key].required &&
       !input_types
     ) {