diff --git a/kairon/shared/chat/models.py b/kairon/shared/chat/models.py
index c0e264529..4205fa099 100644
--- a/kairon/shared/chat/models.py
+++ b/kairon/shared/chat/models.py
@@ -81,6 +81,8 @@ class MessageBroadcastRequest(BaseModel):
     scheduler_config: SchedulerConfiguration = None
     recipients_config: RecipientsConfiguration = None
     template_config: List[TemplateConfiguration] = None
+    template_name: str = None
+    language_code: str = None
     pyscript: str = None
 
     @root_validator
diff --git a/tests/unit_test/events/scheduler_test.py b/tests/unit_test/events/scheduler_test.py
index 9229202b2..d6fa0d682 100644
--- a/tests/unit_test/events/scheduler_test.py
+++ b/tests/unit_test/events/scheduler_test.py
@@ -60,9 +60,7 @@ def test_add_schedule(self, mock_add_job, mock_scheduler, mock_collection):
             assert args[2]
 
             assert args[3][0] == 'message_broadcast'
-            assert args[3][1] == 'Event'
-            assert args[3][2] == {'bot': 'test_bot', 'user': 'test_user', 'event_id': event_id}
-
+            assert args[3][1] == {'bot': 'test_bot', 'user': 'test_user', 'event_id': event_id}
 
     @patch("apscheduler.schedulers.background.BackgroundScheduler.get_jobs", autospec=True)
     @patch("apscheduler.schedulers.background.BackgroundScheduler.start", autospec=True)