At the end of the day if it was officiated correctly from the get go it probably should have been chiefs ball, BUT...
The official came in and called a false start which means the play is dead. And watching it live you can literally hear him blow the whistle while the ball was still free.
The refs realized their mistake, and changed it to an illegal shift to try to correct it. That's not right