Fix the screenshot bug