Add TODO and assert for unhandled cases