fix restart does not works issue