diff --git a/pyproject.toml b/pyproject.toml
index 8b7bf0e..092c4bf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,13 +1,13 @@
 [project]
 name = "uipath-mcp"
-version = "0.0.101"
+version = "0.0.102"
 description = "UiPath MCP SDK"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"
 dependencies = [
     "mcp==1.11.0",
     "pysignalr==1.3.0",
-    "uipath>=2.0.78, <2.1.0",
+    "uipath>=2.1.0, <2.2.0",
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",
diff --git a/src/uipath_mcp/_cli/_runtime/_runtime.py b/src/uipath_mcp/_cli/_runtime/_runtime.py
index 10a7afd..ebc7c30 100644
--- a/src/uipath_mcp/_cli/_runtime/_runtime.py
+++ b/src/uipath_mcp/_cli/_runtime/_runtime.py
@@ -365,7 +365,7 @@ async def _register(self) -> None:
         initialization_successful = False
         tools_result: Optional[ListToolsResult] = None
         server_stderr_output = ""
-        env_vars = dict(server.env)
+        env_vars = server.env
 
         # if server is Coded, include environment variables
         if self.server_type is UiPathServerType.Coded:
diff --git a/src/uipath_mcp/_cli/cli_run.py b/src/uipath_mcp/_cli/cli_run.py
index a711aaf..e4d6c3a 100644
--- a/src/uipath_mcp/_cli/cli_run.py
+++ b/src/uipath_mcp/_cli/cli_run.py
@@ -17,7 +17,7 @@
 
 
 def mcp_run_middleware(
-    entrypoint: Optional[str], input: Optional[str], resume: bool
+    entrypoint: Optional[str], input: Optional[str], resume: bool, **kwargs
 ) -> MiddlewareResult:
     """Middleware to handle MCP server execution"""
 
@@ -37,6 +37,7 @@ async def execute():
             context.entrypoint = entrypoint
             context.input = input
             context.resume = resume
+            context.debug = kwargs.get("debug", False)
             context.logs_min_level = env.get("LOG_LEVEL", "INFO")
             context.job_id = env.get("UIPATH_JOB_KEY")
             context.folder_key = env.get("UIPATH_FOLDER_KEY")
diff --git a/uv.lock b/uv.lock
index c2f92c7..e5317ef 100644
--- a/uv.lock
+++ b/uv.lock
@@ -1996,7 +1996,7 @@ wheels = [
 
 [[package]]
 name = "uipath"
-version = "2.0.78"
+version = "2.1.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "azure-monitor-opentelemetry" },
@@ -2011,9 +2011,9 @@ dependencies = [
     { name = "tomli" },
     { name = "truststore" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/27/7d/b3d25cc9e2094cc4b5a9182aa5b03dfc0c9ee092b590d2e2404057916d47/uipath-2.0.78.tar.gz", hash = "sha256:d8c120bbdea0350050a55ffc61350ee75756136411c70a534ef50b77eb4fa862", size = 1842395 }
+sdist = { url = "https://files.pythonhosted.org/packages/c8/3a/ec026311bf9e0bee34f310c9144337102fb527f129725362764e955f17e4/uipath-2.1.1.tar.gz", hash = "sha256:2a263d2b0c27892321b16a8538cbee6853e41dc14d89ae3ecaa130d9caf3b280", size = 1907270 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/8c/96/496dd5d4f1946eefb51e9c915f20955a104588c87058bf8e7b519b9b4703/uipath-2.0.78-py3-none-any.whl", hash = "sha256:c6328afa36d6a41e21040adbf80457c01f038f435933aedffec7a6986cb82501", size = 133859 },
+    { url = "https://files.pythonhosted.org/packages/4e/1d/8fb6e56901d900d636be27e6b4fddff3d9d3ed19f9371653afdf970f7f73/uipath-2.1.1-py3-none-any.whl", hash = "sha256:4b48d716e557ba6263d13cdde6c9359c21c35eee3bb309990edb0de22cb5da37", size = 136616 },
 ]
 
 [[package]]
@@ -2041,7 +2041,7 @@ dev = [
 requires-dist = [
     { name = "mcp", specifier = "==1.11.0" },
     { name = "pysignalr", specifier = "==1.3.0" },
-    { name = "uipath", specifier = ">=2.0.78,<2.1.0" },
+    { name = "uipath", specifier = ">=2.1.0,<2.2.0" },
 ]
 
 [package.metadata.requires-dev]