added restart behavior