Stress tensor is working