Update to allow restart