diff --git a/package.json b/package.json
index bff2b3a..2496a25 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "getindexify",
-  "version": "0.0.38",
+  "version": "0.0.39",
   "description": "This is the TypeScript client for interacting with the Indexify service.",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
diff --git a/src/client.ts b/src/client.ts
index cceb013..2cf215f 100644
--- a/src/client.ts
+++ b/src/client.ts
@@ -62,26 +62,13 @@ class IndexifyClient {
     return new IndexifyClient(
       serviceUrl,
       namespace,
-      response.data.namespace.extraction_graphs.map(
-        (item: {
-          id: string;
-          name: string;
-          extractor: string;
-          filters_eq: string;
-          input_params: Record<string, string | number>;
-          content_source: string;
-        }) => {
-          // abstraction for filters_eq
-          return {
-            id: item.id,
-            name: item.name,
-            extractor: item.extractor,
-            labels_eq: item.filters_eq,
-            input_params: item.input_params,
-            content_source: item.content_source,
-          };
-        }
-      ) as IExtractionGraph[],
+      response.data.namespace.extraction_graphs.map((graph: { extraction_policies: any[]; }) => ({
+        ...graph,
+        extraction_policies: graph.extraction_policies.map((policy: { filters_eq: any; }) => ({
+          ...policy,
+          labels_eq: policy.filters_eq,  // Transform filters_eq to labels_eq
+        }))
+      })),
       IndexifyClient.getHttpsAgent({ mtlsConfig })
     );
   }
diff --git a/src/types.ts b/src/types.ts
index a11eafb..41140f1 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -71,6 +71,7 @@ export interface IExtractionPolicy {
   labels_eq?: string;
   input_params?: Record<string, string | number>;
   content_source?: string;
+  graph_name: string;
 }
 
 export interface ITaskContentMetadata {
diff --git a/tests/client.test.ts b/tests/client.test.ts
index 7ec03db..bddc342 100644
--- a/tests/client.test.ts
+++ b/tests/client.test.ts
@@ -26,6 +26,7 @@ async function setupExtractionGraph(
   const extractionPolicy: IExtractionPolicy = {
     extractor,
     name: `extractor.${nanoid}`,
+    graph_name: extractionGraphName
   };
   const resp = await client.createExtractionGraph(
     extractionGraphName,