Stable envelope for critical loci