Fix restart

Fix restart