Update the NodeBase type to have a stronger NodeType field that can be undefined even if you specifically define it.