diff --git a/packages/toolbox-core/integration.cloudbuild.yaml b/packages/toolbox-core/integration.cloudbuild.yaml
index 86813f12..ec39c732 100644
--- a/packages/toolbox-core/integration.cloudbuild.yaml
+++ b/packages/toolbox-core/integration.cloudbuild.yaml
@@ -43,4 +43,4 @@ options:
   logging: CLOUD_LOGGING_ONLY
 substitutions:
   _VERSION: '3.13'
-  _TOOLBOX_VERSION: '0.3.0'
+  _TOOLBOX_VERSION: '0.4.0'
diff --git a/packages/toolbox-langchain/integration.cloudbuild.yaml b/packages/toolbox-langchain/integration.cloudbuild.yaml
index 247e1861..ee3a04b2 100644
--- a/packages/toolbox-langchain/integration.cloudbuild.yaml
+++ b/packages/toolbox-langchain/integration.cloudbuild.yaml
@@ -43,4 +43,4 @@ options:
   logging: CLOUD_LOGGING_ONLY
 substitutions:
   _VERSION: '3.13'
-  _TOOLBOX_VERSION: '0.3.0'
+  _TOOLBOX_VERSION: '0.4.0'
diff --git a/packages/toolbox-langchain/tests/test_e2e.py b/packages/toolbox-langchain/tests/test_e2e.py
index 940c01bd..8945657a 100644
--- a/packages/toolbox-langchain/tests/test_e2e.py
+++ b/packages/toolbox-langchain/tests/test_e2e.py
@@ -126,8 +126,8 @@ async def test_run_tool_no_auth(self, toolbox):
             "get-row-by-id-auth",
         )
         with pytest.raises(
-            ToolException,
-            match="{'status': 'Unauthorized', 'error': 'tool invocation not authorized. Please make sure your specify correct auth headers'}",
+            PermissionError,
+            match="Tool get-row-by-id-auth requires authentication, but no valid authentication sources are registered. Please register the required sources before use.",
         ):
             await tool.ainvoke({"id": "2"})
 
@@ -266,8 +266,8 @@ def test_run_tool_no_auth(self, toolbox):
             "get-row-by-id-auth",
         )
         with pytest.raises(
-            ToolException,
-            match="{'status': 'Unauthorized', 'error': 'tool invocation not authorized. Please make sure your specify correct auth headers'}",
+            PermissionError,
+            match="Tool get-row-by-id-auth requires authentication, but no valid authentication sources are registered. Please register the required sources before use.",
         ):
             tool.invoke({"id": "2"})