diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index 81a821266..c5623d3fa 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -241,7 +241,7 @@ export default function ParameterComponent({
         !optionalHandle ? (
           <></>
         )
-        : left === true && type === "str" ? (
+        : left === true && type === "dict" ? (
           <div className="mt-2 w-full">
             <DictComponent
               disabled={disabled}
@@ -253,7 +253,7 @@ export default function ParameterComponent({
             />
           </div>
         )
-        : left === true && type === "dict" ? (
+        : left === true && type === "keypair" ? (
           <div className="mt-2 w-full">
             <KeypairListComponent
               disabled={disabled}
diff --git a/src/frontend/src/components/dictComponent/index.tsx b/src/frontend/src/components/dictComponent/index.tsx
index 1269bda17..ef1a31ffc 100644
--- a/src/frontend/src/components/dictComponent/index.tsx
+++ b/src/frontend/src/components/dictComponent/index.tsx
@@ -1,5 +1,5 @@
 import { useEffect } from "react";
-import { KeyPairListComponent } from "../../types/components";
+import { DictComponentType } from "../../types/components";
 
 import { classNames } from "../../utils/utils";
 import { Input } from "../ui/input";
@@ -10,7 +10,7 @@ export default function DictComponent({
   onChange,
   disabled,
   editNode = false,
-}: KeyPairListComponent): JSX.Element {
+}: DictComponentType): JSX.Element {
   useEffect(() => {
     if (disabled) {
       onChange([""]);
diff --git a/src/frontend/src/components/keypairListComponent/index.tsx b/src/frontend/src/components/keypairListComponent/index.tsx
index 87cad64d0..26e6d7f46 100644
--- a/src/frontend/src/components/keypairListComponent/index.tsx
+++ b/src/frontend/src/components/keypairListComponent/index.tsx
@@ -1,5 +1,5 @@
 import { useEffect } from "react";
-import { KeyPairListComponent } from "../../types/components";
+import { KeyPairListComponentType } from "../../types/components";
 
 import _ from "lodash";
 import { TypeModal } from "../../constants/enums";
@@ -14,7 +14,7 @@ export default function KeypairListComponent({
   disabled,
   editNode = false,
   duplicateKey
-}: KeyPairListComponent): JSX.Element {
+}: KeyPairListComponentType): JSX.Element {
   useEffect(() => {
     if (disabled) {
       onChange([""]);
diff --git a/src/frontend/src/modals/EditNodeModal/index.tsx b/src/frontend/src/modals/EditNodeModal/index.tsx
index 85a149db8..437cdf232 100644
--- a/src/frontend/src/modals/EditNodeModal/index.tsx
+++ b/src/frontend/src/modals/EditNodeModal/index.tsx
@@ -206,7 +206,7 @@ const EditNodeModal = forwardRef(
                                     />
                                   )
                                   : myData.node?.template[templateParam]
-                                      .type === "str" ? (
+                                      .type === "dict" ? (
                                     <div className="mt-2 w-full">
                                         <DictComponent
                                           disabled={disabled}
@@ -219,7 +219,7 @@ const EditNodeModal = forwardRef(
                                     </div>
                                   )
                                   : myData.node?.template[templateParam]
-                                      .type === "dict" ? (
+                                      .type === "keypair" ? (
                                     <div className="mt-2 w-full">
                                       <KeypairListComponent
                                         disabled={disabled}
diff --git a/src/frontend/src/types/components/index.ts b/src/frontend/src/types/components/index.ts
index 9b6a78aa6..bebcbefe5 100644
--- a/src/frontend/src/types/components/index.ts
+++ b/src/frontend/src/types/components/index.ts
@@ -54,7 +54,7 @@ export type InputListComponentType = {
   editNode?: boolean;
 };
 
-export type KeyPairListComponent = {
+export type KeyPairListComponentType = {
   value: any;
   onChange: (value: string[]) => void;
   disabled: boolean;
@@ -62,6 +62,13 @@ export type KeyPairListComponent = {
   duplicateKey?: boolean;
 };
 
+export type DictComponentType = {
+  value: any;
+  onChange: (value: string[]) => void;
+  disabled: boolean;
+  editNode?: boolean;
+};
+
 export type TextAreaComponentType = {
   field_name?: string;
   nodeClass?: APIClassType;