diff --git a/packages/ui/src/elements/Menu.tsx b/packages/ui/src/elements/Menu.tsx
index 0579758d082..da7b1bc9919 100644
--- a/packages/ui/src/elements/Menu.tsx
+++ b/packages/ui/src/elements/Menu.tsx
@@ -58,7 +58,6 @@ export const MenuTrigger = (props: MenuTriggerProps) => {
   }
 
   return cloneElement(children, {
-    // @ts-expect-error
     ref: reference,
     elementDescriptor: children.props.elementDescriptor || descriptors.menuButton,
     elementId: children.props.elementId || descriptors.menuButton.setId(elementId),
diff --git a/packages/ui/tsconfig.json b/packages/ui/tsconfig.json
index c97f2d88d06..caa880cc04e 100644
--- a/packages/ui/tsconfig.json
+++ b/packages/ui/tsconfig.json
@@ -27,7 +27,8 @@
       // in order to make git merges easier
       // TODO: @nikos remove after v6 release
       "@/ui*": ["./src/*"]
-    }
+    },
+    "types": ["@rspack/core/module"]
   },
   "exclude": [
     "node_modules",