changed restart a little bit