Fix layout when IR1 has several root nodes