diff --git a/src/components/ImageViewer.tsx b/src/components/ImageViewer.tsx index 6fe4111..5ffd310 100644 --- a/src/components/ImageViewer.tsx +++ b/src/components/ImageViewer.tsx @@ -102,11 +102,11 @@ function ImageViewer(props: ImageViewerProps, ref: ForwardedRef) { setWithAnimation(true); if (isFavorited) { - setFavorites(favorites.filter(fav => fav?.id !== media[currentIndex].id)); + setFavorites(prev => prev.filter(fav => fav?.id !== media[currentIndex].id)); } else { - setFavorites([...favorites, media[currentIndex]]); + setFavorites(prev => [...prev, media[currentIndex]]); } - }, [favorites, media, currentIndex, isFavorited, setFavorites]); + }, [media, currentIndex, isFavorited, setFavorites]); return (