diff --git a/app/client/src/modules/ui-builder/ui/wds/WDSCustomWidget/component/index.tsx b/app/client/src/modules/ui-builder/ui/wds/WDSCustomWidget/component/index.tsx
index 9b0e7d4523be..93c7cedabb7a 100644
--- a/app/client/src/modules/ui-builder/ui/wds/WDSCustomWidget/component/index.tsx
+++ b/app/client/src/modules/ui-builder/ui/wds/WDSCustomWidget/component/index.tsx
@@ -80,7 +80,7 @@ export function CustomWidgetComponent(props: CustomWidgetComponentProps) {
   // the iframe can make changes to the model, when it needs to
   // this is done by sending a CUSTOM_WIDGET_UPDATE_MODEL message to the parent window
   const handleModelUpdate = (message: Record<string, unknown>) => {
-    onUpdateModel(message.model as Record<string, unknown>);
+    onUpdateModel(message as Record<string, unknown>);
   };
 
   // the iframe elements can trigger events. Triggered events here would mean