fix restart button