Long post ahead but the TLDR is that this should be a story of redemption; KO needs to redeem himself to Sami for his past mistakes, Sami needs to redeem himself to Jey Uso for causing a rift between him and his Bloodline and Jey needs to redeem himself with......himself.
I'm not going to lie, I loved the hug, I just don't like how they got to it. Sami should have been playing Kevin's role. It should have been on sight as soon as KO showed from selling that beatdown. For all of the betrayals, manipulations, beatings, powerbombs on the apron, for constantly putting him in the position where he had to choose between people who really showed genuine care for him (more so than Owens ever did) and a serial backstabber he somehow loves like a brother and for not even being there to help him after the obvious consequences. It should have been Sami attacking Owens on sight, getting the two in a ring for a promo where he lays out his feelings of anger and we set in motion KO trying to convince Sami to help him take down the Bloodline and everyone wondering if he is just using him yet again to further his own agenda.
Jey Uso should have always had this angry energy with Sami. It should have never really been a question on if he was still down with the Bloodline. His conflict would be how much he feels Sami deserves all the worst that can come to him but only at his hands while trying to tacitly protect him from the wrath of the Bloodline as a whole. Every Bloodline beatdown on Zayn should have featured Jey being overzealous with his thirst for revenge; he wants exclusive rights to hurt Sami. But of course Paul Heyman, the wily Jew that he is, notices that Jey might be going a bit too easy on "Schmoolie," something Reigns was already considering. Of course he leans on Jimmy to figure out what's going on with the usual gaslighting and threats which would culminate in a similar Royal Rumble "test." The Bloodline corner Zayn and beat his ass (after sending KO to a local "medical facility") and this time Jey has to watch and wait his turn. After a brutal beating it's his time to make the choice. And of course he chooses his family. But you could see through the beating he was pained and adding to the damage already done hurts him as well, but he's still angry because ultimately Sami betrayed him and his family for someone who would never do the same for Sami and made Jey look like an idiot in the process. Throughout this part of the story Sami is extremely hesitant to put up a fight against Jey because he feels guilty and knows that Jey has a legitimate argument.
In the end KO would limp his fat ass out there, fight a losing battle against the Bloodline and end up getting laid out alongside Sami but that would cement the WM match right there; everyone has made their choice. KO will support Sami even if Sami refuses to trust him. Sami decides that this is a different KO and Jey has ultimately made his choice so the gloves have to come off. Why not take down the Bloodline with the devil you know? Even though Jey made his choice by beating the dead horse that was Sami Zayn, seeing KO come to his aid even after constantly being distrusted and rundown for his past actions, Sami still embraced him in the end showing that their is a way out and you're not cursed to be the worst version of yourself forever. You can redeem past mistakes.