diff --git a/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py b/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py
index 457c79623..16d439169 100644
--- a/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py
+++ b/packages/jupyter-ai/jupyter_ai/tests/test_handlers.py
@@ -8,6 +8,7 @@
 from jupyter_ai.chat_handlers import DefaultChatHandler, learn
 from jupyter_ai.config_manager import ConfigManager
 from jupyter_ai.handlers import RootChatHandler
+from jupyter_ai.history import BoundedChatHistory
 from jupyter_ai.models import (
     AgentStreamChunkMessage,
     AgentStreamMessage,
@@ -18,7 +19,6 @@
     PendingMessage,
     Persona,
 )
-from jupyter_ai.history import BoundedChatHistory
 from jupyter_ai_magics import BaseProvider
 from langchain_community.llms import FakeListLLM
 from tornado.httputil import HTTPServerRequest