diff --git a/StabilityMatrix.Core/Inference/ComfyClient.cs b/StabilityMatrix.Core/Inference/ComfyClient.cs
index 9e7ab6bf..3f3b894e 100644
--- a/StabilityMatrix.Core/Inference/ComfyClient.cs
+++ b/StabilityMatrix.Core/Inference/ComfyClient.cs
@@ -471,8 +471,9 @@ public class ComfyClient : InferenceClientBase
     {
         var response = await comfyApi.GetObjectInfo(nodeName, cancellationToken).ConfigureAwait(false);
 
-        var info = response[nodeName];
-        return info.Input.GetOptionalValueAsNestedList(optionName);
+        var info = response.GetValueOrDefault(nodeName);
+
+        return info?.Input.GetOptionalValueAsNestedList(optionName);
     }
 
     protected override void Dispose(bool disposing)