Fixed screenshot bug