diff --git a/libs/sdk-js/package.json b/libs/sdk-js/package.json
index f8c9c5a1a..2be6a6a77 100644
--- a/libs/sdk-js/package.json
+++ b/libs/sdk-js/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-sdk",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "description": "Client library for interacting with the LangGraph API",
   "type": "module",
   "packageManager": "yarn@1.22.19",
diff --git a/libs/sdk-js/src/client.mts b/libs/sdk-js/src/client.mts
index 642832eed..a39569a70 100644
--- a/libs/sdk-js/src/client.mts
+++ b/libs/sdk-js/src/client.mts
@@ -427,15 +427,18 @@ export class ThreadsClient extends BaseClient {
   async updateState<ValuesType = DefaultValues>(
     threadId: string,
     options: { values: ValuesType; checkpointId?: string; asNode?: string },
-  ): Promise<void> {
-    return this.fetch<void>(`/threads/${threadId}/state`, {
-      method: "POST",
-      json: {
-        values: options.values,
-        checkpoint_id: options.checkpointId,
-        as_node: options?.asNode,
+  ): Promise<Pick<Config, "configurable">> {
+    return this.fetch<Pick<Config, "configurable">>(
+      `/threads/${threadId}/state`,
+      {
+        method: "POST",
+        json: {
+          values: options.values,
+          checkpoint_id: options.checkpointId,
+          as_node: options?.asNode,
+        },
       },
-    });
+    );
   }
 
   /**