fixed restart issue