more restart fix

more restart fix