Issue with reflex reactflow

[narenjcs]

Path: /docs/wrapping-react/example/

not able to add custom node_type

node_type = {"colorPicker": rx.vstack(color_picker_node())}

return rx.vstack(
    color_picker_node(),
    react_flow(
        background(
            color="green",
        ),
        controls(),
        mini_map(),
        nodes_draggable=True,
        nodes_connectable=True,
        on_connect=lambda e0: State.on_connect(e0),
        on_nodes_change=lambda e0: State.on_nodes_change(
            e0
        ),
        nodes=State.nodes,
        edges=State.edges,
        node_types=node_type,
        # fit_view=True,
    ))