I find medium firm tofu holds together better when cooking in sauces like this. It’s less brittle than firm tofu and has some give under pressure so it bends/squishes instead of crumbling.