litte update to allow restart