diff --git a/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py b/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py
index d2e73ce6c..2ed6c7755 100644
--- a/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py
+++ b/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py
@@ -5,12 +5,11 @@
 from unittest import mock
 
 import pytest
+from jupyter_ai.extension import DEFAULT_HELP_MESSAGE_TEMPLATE
 from jupyter_ai.chat_handlers import DefaultChatHandler, learn
 from jupyter_ai.config_manager import ConfigManager
 from jupyter_ai.handlers import RootChatHandler
 from jupyter_ai.models import (
-    AgentStreamChunkMessage,
-    AgentStreamMessage,
     ChatClient,
     ClosePendingMessage,
     HumanChatMessage,
@@ -73,6 +72,8 @@ def broadcast_message(message: Message) -> None:
             root_dir="",
             preferred_dir="",
             dask_client_future=None,
+            help_message_template=DEFAULT_HELP_MESSAGE_TEMPLATE,
+            chat_handlers={}
         )