diff --git a/jupyter_ai_magic_commands/magics.py b/jupyter_ai_magic_commands/magics.py
index ed16a5f..2aa9e7b 100644
--- a/jupyter_ai_magic_commands/magics.py
+++ b/jupyter_ai_magic_commands/magics.py
@@ -15,7 +15,7 @@
 # from jupyter_ai.model_providers.model_list import CHAT_MODELS
 from jupyter_ai_litellm.model_list import CHAT_MODELS
 
-from ._version import __version__
+from . import __version__
 from .parsers import (
     CellArgs,
     DeleteArgs,