Error when using menu and listbox with child function and fragment in React 19

[isoroka-plana]

What package within Headless UI are you using?

@headlessui/react

What version of that package are you using?

For example: v2.2.0

What browser are you using?

For example: Chrome

Reproduction URL

https://codesandbox.io/p/devbox/funny-wood-jtchxm

Describe your issue

After updating to React 19 following error has begun to appear in my console:

 Invalid prop `data-headlessui-state` supplied to `React.Fragment`. React.Fragment can only have `key` and `children` props. 

When direct child of Menu or Listbox elements is a react fragment (my use case is that I am using a function a function for children to access the component's APIs).
Example:

<Menu>
  ({open}) => (
    <>
      <MenuButton>Text</MenuButton>
      {open && <MenuItems></MenuItems}
    </>
  )
</Menu>

There may be other components affected, I did not test.