diff --git a/app/client/src/ce/constants/ModuleConstants.ts b/app/client/src/ce/constants/ModuleConstants.ts
index 189b5a014aaa..856db60531e6 100644
--- a/app/client/src/ce/constants/ModuleConstants.ts
+++ b/app/client/src/ce/constants/ModuleConstants.ts
@@ -4,8 +4,8 @@ type ID = string;
 
 export enum MODULE_TYPE {
   QUERY = "QUERY_MODULE",
-  JS = "JS_MODULE",
-  UI = "UI_MODULE",
+  JS = "JS",
+  UI = "UI",
 }
 
 export interface ModuleInput {
@@ -17,8 +17,7 @@ export interface ModuleInputSection {
   children?: ModuleInput[];
 }
 
-export interface Module
-  extends Pick<ModuleMetadata, "pluginId" | "pluginType" | "datasourceId"> {
+export interface Module extends ModuleMetadata {
   id: ID;
   name: string;
   packageId: ID;