add restart behavior