fixed restart problem