Correct stress tensor calculation