[ENH] UFEM: Use the full viscous stress matrix