This looks ok, but I really having real functionality inside a WARN_ON statement as that obsufcateѕ things. But I don't think we even really need the WARN_ON here at all.