He only wanted to punish him for betrayal if you didn't battle Taka in the water treatment plant.
He said Taka would be punished for incompetence the other way which might still be death but is more likely to be something less severe (except this is Solaris so who knows).