maybe fix restart?