Fixed virial calculation (wrong formula)