Add restart magic