diff --git a/client/src/components/GCode3DImageViewer.vue b/client/src/components/GCode3DImageViewer.vue
index b5af0c0..a6f670c 100644
--- a/client/src/components/GCode3DImageViewer.vue
+++ b/client/src/components/GCode3DImageViewer.vue
@@ -110,7 +110,7 @@ onMounted(async () => {
     // Clean up when the modal is hidden
     if (preview) {
       preview.clear();
-      setPreview(null);
+      preview = null;
     }
   });
 });
@@ -118,7 +118,7 @@ onMounted(async () => {
 onUnmounted(() => {
   if (preview) {
     preview.clear();
-    setPreview(null);
+    preview = null;
   }
 });