diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
index 09b64d599..df1821626 100644
--- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
+++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -27,7 +27,6 @@
     QianfanChatEndpoint,
 )
 from langchain.chat_models.base import BaseChatModel
-from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.llms import (
     AI21,
     Anthropic,
@@ -46,6 +45,7 @@
 from langchain.pydantic_v1 import BaseModel, Extra, root_validator
 from langchain.schema import LLMResult
 from langchain.utils import get_from_dict_or_env
+from langchain_google_genai import ChatGoogleGenerativeAI
 
 
 class EnvAuthStrategy(BaseModel):