Fixes to Tree Assignment And Lrx