- Add afterRoot support