fixed the restart issue