Updates for late and early fusion!