mask predictions not shown when full screen image in media view

When viewing the dashboard I am logging some images with the model mask prediction on them. As seen in the screen shot I can view the images and masks 1 at a time when shown in the default layout. However when I click the image to full screen it, the mask disappears and it only shows the image. When I click the image to then download the mask appears in the downloaded file. How do I make it so when I click the image in the wandb dashboard I still see the image in full screen view. Here is the code I use:

                    "prediction": wandb.Image(
                        input_image,
                        masks={
                            "predictions": {
                                "mask_data": instance_mask_resized,
                                "class_labels": class_labels
                            }
                        },
                        caption=f"Instance Segmentation - Iter {self.iter}"
                    ),
                    "iteration": self.iter
                })