Consider the Navier-Stokes equations instead of Euler equations,
and we define the shear-energy tensor as
With the  condition that the normal component of the total stress on the free surface must vanish, i.e., , where  is the outward unit normal, the second order virial equation then be
This can be easily extended to rotating frame of references.