Quick fix for screenshot_error KeyError