add restart message handler