They probably saw the backlash on the forum, and elsewhere, and are trying to change their story.
Good na, if she is alive, and they show her contributing to the ending, either by catching Roshan or by finding an antidote for Virus.
That would show her as heroic, befitting her role.
Showing Jai shooting Shivani was the silliest track for this season, anyways. That didn't paint too good a picture of the top ATU professionals they are.
It is ok to say that they are ready to sacrifice their life for the country, etc.
But in this story, it didn't come across as a sacrifice, as that kind of foundation was not set by the story.
It just came across as plain silly of the two top professionals to do something like because they could not think of anything better.