diff --git a/src/frontend/src/controllers/API/index.ts b/src/frontend/src/controllers/API/index.ts
index 37e579428..8e5517131 100644
--- a/src/frontend/src/controllers/API/index.ts
+++ b/src/frontend/src/controllers/API/index.ts
@@ -357,7 +357,11 @@ export async function postCustomComponent(
   code: string,
   apiClass: APIClassType
 ): Promise<AxiosResponse<APIClassType>> {
-  return await api.post(`${BASE_URL_API}custom_component`, { code });
+  let template = apiClass.template
+  return await api.post(`${BASE_URL_API}custom_component`, {
+    code,
+    template
+  });
 }
 
 export async function postCustomComponentUpdate(