Tree shaking is broken

[grefrit]

What package within Headless UI are you using?
@headlessui/react

What version of that package are you using?
v1.5.0

What browser are you using?
Chrome

Reproduction URL

1. Clone https://github.com/GrefriT/headlessui-bug
2. yarn
3. yarn build
   It will generate bundle map

Describe your issue
Hello. It seems that version 1.5.0 broke tree shaking. I'm using headlessui in NextJS project

Also I see that bundlephobia can't build this package

[RobinMalfait]

This should be fixed by #1247, and will be available in the next release.

You can already try it using:

• npm install @headlessui/react@insiders or yarn add @headlessui/react@insiders.
• npm install @headlessui/vue@insiders or yarn add @headlessui/vue@insiders.

Before:

After:

[grefrit]

Amazing, butt... I can't confirm it's working or not, because of this Type error:

Type '(<TTag extends ElementType<any> = "button">(props: Props<TTag, SwitchRenderPropArg, "name" | "value" | "onChange" | "checked" | SwitchPropsWeControl> & { ...; }, ref: Ref<...>) => Element | null) & { 
...; } & { ...; }' does not satisfy the constraint 'JSXElementConstructor<any> | keyof IntrinsicElements'.
  Type '(<TTag extends ElementType<any> = "button">(props: Props<TTag, SwitchRenderPropArg, "name" | "value" | "onChange" | "checked" | SwitchPropsWeControl> & { ...; }, ref: Ref<...>) => Element | null) & { ...; } & { ...; }' is not assignable to type '(props: any) => ReactElement<any, any> | null'.

I'm using ComponentProps<typeof Switch> to extract types (ComponentProps imported from 'react'). Probably, headlessui should export component props, because for example, I'm building my own components on top of headlessui component for UI consistency