diff --git a/apps/web/src/windows/ImageWindow.tsx b/apps/web/src/windows/ImageWindow.tsx
index 428e9ec..121f853 100644
--- a/apps/web/src/windows/ImageWindow.tsx
+++ b/apps/web/src/windows/ImageWindow.tsx
@@ -10,9 +10,7 @@ export const ImageWindow = observer(
         const [imageSrc, setImageSrc] = useState<string>();
 
         useEffect(() => {
-            const subscription = node.outputs.output.subscribe(url => {
-                setImageSrc(url);
-            });
+            const subscription = node.outputs.output.subscribe(setImageSrc);
 
             return () => {
                 subscription.unsubscribe();