Updates to fix the screenshot image