fix restart prompt