From aa7fe577f0ab5efba47e877695f932e144cb321e Mon Sep 17 00:00:00 2001
From: Jonathan Tran <jonnytran@gmail.com>
Date: Wed, 29 Jan 2025 20:11:00 -0500
Subject: [PATCH] Add updating the artifact graph after mock execution

---
 src/lang/KclSingleton.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/lang/KclSingleton.ts b/src/lang/KclSingleton.ts
index 35294ba4dd..c452a2a6cc 100644
--- a/src/lang/KclSingleton.ts
+++ b/src/lang/KclSingleton.ts
@@ -452,6 +452,7 @@ export class KclManager {
       this.lastSuccessfulProgramMemory = execState.memory
       this.lastSuccessfulOperations = execState.operations
     }
+    this.engineCommandManager.updateArtifactGraph(execState.artifactGraph)
   }
   cancelAllExecutions() {
     this._cancelTokens.forEach((_, key) => {