Adding in image fix