diff --git a/src/frontend/src/controllers/API/index.ts b/src/frontend/src/controllers/API/index.ts
index 6b2517c83..165ce843c 100644
--- a/src/frontend/src/controllers/API/index.ts
+++ b/src/frontend/src/controllers/API/index.ts
@@ -338,11 +338,4 @@ export async function uploadFile(
   const formData = new FormData();
   formData.append("file", file);
   return await axios.post(`/api/v1/upload/${id}`, formData);
-}
-
-export async function postCustomComponent(
-  code: string,
-  apiClass: APIClassType
-): Promise<AxiosResponse<APIClassType>> {
-  return await axios.post(`/api/v1/custom_component`, { code });
-}
+}
\ No newline at end of file
diff --git a/src/frontend/src/modals/codeAreaModal/index.tsx b/src/frontend/src/modals/codeAreaModal/index.tsx
index 7b48fe6da..02c9ff7c5 100644
--- a/src/frontend/src/modals/codeAreaModal/index.tsx
+++ b/src/frontend/src/modals/codeAreaModal/index.tsx
@@ -7,7 +7,7 @@ import "ace-builds/src-noconflict/theme-twilight";
 import "ace-builds/src-noconflict/ext-language_tools";
 // import "ace-builds/webpack-resolver";
 import { darkContext } from "../../contexts/darkContext";
-import { postCustomComponent, postValidateCode } from "../../controllers/API";
+import { postValidateCode } from "../../controllers/API";
 import { alertContext } from "../../contexts/alertContext";
 import {
   Dialog,
@@ -63,7 +63,8 @@ export default function CodeAreaModal({
             setSuccessData({
               title: "Code is ready to run",
             });
-            // setValue(code);
+            setValue(code);
+            setModalOpen(false);
           } else {
             if (funcErrors.length !== 0) {
               setErrorData({
@@ -90,13 +91,6 @@ export default function CodeAreaModal({
           title: "There is something wrong with this code, please review it",
         });
       });
-    postCustomComponent(code, nodeClass).then((apiReturn) => {
-      const data = apiReturn.data;
-      if (data) {
-        setNodeClass(data);
-        setModalOpen(false);
-      }
-    });
   }
 
   return (