[TASK] Small clean up in the page tree code