bug fix restart