fix restart issue