[M] : modify the register tree, better