Small fix to image