diff --git a/pyproject.toml b/pyproject.toml
index c07b54f..21b5add 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "mcp-server-jupyter"
-version = "0.1.5"
+version = "0.1.6"
 description = "MCP server for Jupyter notebooks"
 readme = "README.md"
 requires-python = ">=3.10"
diff --git a/src/mcp_server_jupyter/__init__.py b/src/mcp_server_jupyter/__init__.py
index 056efee..c832bc9 100644
--- a/src/mcp_server_jupyter/__init__.py
+++ b/src/mcp_server_jupyter/__init__.py
@@ -1 +1,2 @@
 from .server import main as main
+from .server import run as run