modify restart method.