You can simplify but it’s still unclear if the U^2 term you derived was supposed to be U^1 (before the simplification, I.e. you did an extra multiplication). It just makes bare type checking slightly more ambiguous.