Bug fix with image type