You think that
would've been a twist?
The film uses one specific element from TWoK, that's hardly a whole act.
A better punishment would have been to have him be sent to a rehabilitation colony, where (if
the technology exists) his personality is replaced with a new one, as is done on Babylon 5