fix restart test