diff --git a/src/lib/components/ComfyApp.svelte b/src/lib/components/ComfyApp.svelte
index f216d9b..40e81a7 100644
--- a/src/lib/components/ComfyApp.svelte
+++ b/src/lib/components/ComfyApp.svelte
@@ -79,7 +79,7 @@
      for (let i = 0; i < serializedGraph.nodes.length; i++) {
          let serializedNode = serializedGraph.nodes[i];
          let frontendWidgetStates = frontendState[serializedNode.id];
-         if (frontendWidgetStates) {
+         if (frontendWidgetStates && serializedNode.widgets_values) {
              for (let j = 0; j < serializedNode.widgets_values.length; j++) {
                  let frontendWidgetState = frontendWidgetStates[j];
 
diff --git a/src/lib/components/ComfyPane.svelte b/src/lib/components/ComfyPane.svelte
index 56d4b6e..e4c55ba 100644
--- a/src/lib/components/ComfyPane.svelte
+++ b/src/lib/components/ComfyPane.svelte
@@ -46,7 +46,7 @@
 
  onDestroy(unsubscribe);
 
- $: if ($queueState.runningNodeId) {
+ $: if ($queueState) {
      for (let dragItem of dragItems) {
          dragItem.isNodeExecuting = $queueState.runningNodeId === dragItem.node.id;
      }
diff --git a/src/lib/components/LightboxModal.svelte b/src/lib/components/LightboxModal.svelte
index 2c7387f..d52ebd6 100644
--- a/src/lib/components/LightboxModal.svelte
+++ b/src/lib/components/LightboxModal.svelte
@@ -51,6 +51,10 @@
      flex-direction: column;
  }
 
+ #modalImage {
+     overflow: hidden;
+ }
+
  .modalControls {
      display: flex;
      gap: 1em;