Fixed U_sq bug