diff --git a/src/frontend/src/customNodes/genericNode/components/outputModal/components/switchOutputView/components/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/components/index.tsx
similarity index 100%
rename from src/frontend/src/customNodes/genericNode/components/outputModal/components/switchOutputView/components/index.tsx
rename to src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/components/index.tsx
diff --git a/src/frontend/src/customNodes/genericNode/components/outputModal/components/switchOutputView/helpers/convert-to-table-rows.ts b/src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/helpers/convert-to-table-rows.ts
similarity index 100%
rename from src/frontend/src/customNodes/genericNode/components/outputModal/components/switchOutputView/helpers/convert-to-table-rows.ts
rename to src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/helpers/convert-to-table-rows.ts
diff --git a/src/frontend/src/customNodes/genericNode/components/outputModal/components/switchOutputView/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/index.tsx
similarity index 100%
rename from src/frontend/src/customNodes/genericNode/components/outputModal/components/switchOutputView/index.tsx
rename to src/frontend/src/CustomNodes/GenericNode/components/outputModal/components/switchOutputView/index.tsx
diff --git a/src/frontend/src/customNodes/genericNode/components/outputModal/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/outputModal/index.tsx
similarity index 100%
rename from src/frontend/src/customNodes/genericNode/components/outputModal/index.tsx
rename to src/frontend/src/CustomNodes/GenericNode/components/outputModal/index.tsx
diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index be62de826..719bd9ab4 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -21,7 +21,7 @@ import {
   LANGFLOW_SUPPORTED_TYPES,
   TOOLTIP_EMPTY,
 } from "../../../../constants/constants";
-import OutputModal from "../../../../customNodes/genericNode/components/outputModal";
+import OutputModal from "../outputModal";
 import { Case } from "../../../../shared/components/caseComponent";
 import useFlowStore from "../../../../stores/flowStore";
 import useFlowsManagerStore from "../../../../stores/flowsManagerStore";
@@ -430,7 +430,7 @@ export default function ParameterComponent({
                       });
                     }}
                     name={name}
-                    data={data.node?.template[name]}
+                    data={data.node?.template[name]!}
                   />
                 </div>
                 {data.node?.template[name]?.refresh_button && (
diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index 0dd8d9694..63b84a847 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -12,8 +12,8 @@ import {
   STATUS_BUILDING,
 } from "../../constants/constants";
 import { BuildStatus } from "../../constants/enums";
-import { countHandlesFn } from "../../customNodes/helpers/count-handles";
-import { getSpecificClassFromBuildStatus } from "../../customNodes/helpers/get-class-from-build-status";
+import { countHandlesFn } from "../helpers/count-handles";
+import { getSpecificClassFromBuildStatus } from "../helpers/get-class-from-build-status";
 import NodeToolbarComponent from "../../pages/FlowPage/components/nodeToolbarComponent";
 import useAlertStore from "../../stores/alertStore";
 import { useDarkStore } from "../../stores/darkStore";
diff --git a/src/frontend/src/customNodes/helpers/count-handles.ts b/src/frontend/src/CustomNodes/helpers/count-handles.ts
similarity index 100%
rename from src/frontend/src/customNodes/helpers/count-handles.ts
rename to src/frontend/src/CustomNodes/helpers/count-handles.ts
diff --git a/src/frontend/src/customNodes/helpers/get-class-from-build-status.ts b/src/frontend/src/CustomNodes/helpers/get-class-from-build-status.ts
similarity index 100%
rename from src/frontend/src/customNodes/helpers/get-class-from-build-status.ts
rename to src/frontend/src/CustomNodes/helpers/get-class-from-build-status.ts
diff --git a/src/frontend/src/types/components/index.ts b/src/frontend/src/types/components/index.ts
index 484ed5c84..18bba4c2e 100644
--- a/src/frontend/src/types/components/index.ts
+++ b/src/frontend/src/types/components/index.ts
@@ -84,7 +84,7 @@ export type InputGlobalComponentType = {
   onChange: (value: string) => void;
   setDb: (value: boolean) => void;
   name: string;
-  data: NodeDataType;
+  data: TemplateVariableType;
   editNode?: boolean;
   playgroundDisabled?: boolean;
 };