diff --git a/examples/SampleApp/src/screens/ChannelImagesScreen.tsx b/examples/SampleApp/src/screens/ChannelImagesScreen.tsx
index 058cb92f05..bf4bfe82eb 100644
--- a/examples/SampleApp/src/screens/ChannelImagesScreen.tsx
+++ b/examples/SampleApp/src/screens/ChannelImagesScreen.tsx
@@ -66,7 +66,11 @@ export const ChannelImagesScreen: React.FC<ChannelImagesScreenProps> = ({
     params: { channel },
   },
 }) => {
-  const { images, setImage, setImages } = useImageGalleryContext<StreamChatGenerics>();
+  const {
+    messages: images,
+    setMessages: setImages,
+    setSelectedMessage: setImage,
+  } = useImageGalleryContext<StreamChatGenerics>();
   const { setOverlay } = useOverlayContext();
   const { loading, loadMore, messages } = usePaginatedAttachments(channel, 'image');
   const {