Fix restart error