Masks not aligned with image

Went through the notebook to generate embeddings for dogs.jpg. However, when hovering over the image the mask is offset above the original image. I have tried this with various images and the mask is always offset or skewed in some way. I tried running through the demo on a clean repo and still ran into this issue. I suspect it has something to with the scaled down model being displayed (despite console.log telling me that image and maskImage are the same dimensions). Anyone have any pointers?