fix screenshot restart