Yeah, I really wish they had kept Keith as a traitor. I was so fucking stoked when the betrayal happened because I thought they were pulling a Julius again, but then he reforms in the next mission and you never see him again until the end of the game during the dance sequence.