diff --git a/packages/@headlessui-react/src/components/switch/switch.tsx b/packages/@headlessui-react/src/components/switch/switch.tsx
index 9e07285..71e6b31 100644
--- a/packages/@headlessui-react/src/components/switch/switch.tsx
+++ b/packages/@headlessui-react/src/components/switch/switch.tsx
@@ -111,7 +111,7 @@ let SwitchRoot = forwardRefWithAs(function Switch<
     name?: string
     value?: string
   },
-  ref: Ref<HTMLElement>
+  ref: Ref<HTMLButtonElement>
 ) {
   let internalId = useId()
   let {
@@ -128,7 +128,6 @@ let SwitchRoot = forwardRefWithAs(function Switch<
   let switchRef = useSyncRefs(
     internalSwitchRef,
     ref,
-    // @ts-expect-error figure out the correct type here
     groupContext === null ? null : groupContext.setSwitch
   )