diff --git a/dist/index.js b/dist/index.js
index ab90068..0b7a3b9 100644
--- a/dist/index.js
+++ b/dist/index.js
@@ -43064,8 +43064,8 @@ async function main({
     });
   });
   openPullRequests.forEach((openPullRequest) => {
-    const hasExistingBasePullRequest = repoGraph.hasNode(openPullRequest.base.ref);
-    if (hasExistingBasePullRequest) {
+    const hasExistingBase = repoGraph.hasNode(openPullRequest.base.ref);
+    if (hasExistingBase) {
       repoGraph.mergeDirectedEdge(openPullRequest.base.ref, openPullRequest.head.ref);
       return;
     }
@@ -43112,7 +43112,7 @@ async function main({
         stackGraph2.dropNode(ref);
       }
     });
-    return import_graphology.DirectedGraph.from(stackGraph2.toJSON());
+    return stackGraph2;
   };
   const getOutput = (graph) => {
     const lines = [];
diff --git a/src/main.ts b/src/main.ts
index 607ffaf..c54088f 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -40,8 +40,8 @@ export async function main({
   })
 
   openPullRequests.forEach((openPullRequest) => {
-    const hasExistingBasePullRequest = repoGraph.hasNode(openPullRequest.base.ref)
-    if (hasExistingBasePullRequest) {
+    const hasExistingBase = repoGraph.hasNode(openPullRequest.base.ref)
+    if (hasExistingBase) {
       repoGraph.mergeDirectedEdge(openPullRequest.base.ref, openPullRequest.head.ref)
 
       return
@@ -98,7 +98,7 @@ export async function main({
       }
     })
 
-    return DirectedGraph.from(stackGraph.toJSON())
+    return stackGraph
   }
 
   const getOutput = (graph: DirectedGraph<StackNodeAttributes>) => {