diff --git a/code/ui/blocks/src/blocks/useOf.ts b/code/ui/blocks/src/blocks/useOf.ts
index 8e5d9d5f7a4f..8c2c55310029 100644
--- a/code/ui/blocks/src/blocks/useOf.ts
+++ b/code/ui/blocks/src/blocks/useOf.ts
@@ -1,6 +1,5 @@
 import type {
   DocsContextProps,
-  ModuleExport,
   ResolvedModuleExportType,
   ResolvedModuleExportFromType,
 } from '@storybook/types';
@@ -16,7 +15,7 @@ export type Of = Parameters<DocsContextProps['resolveOf']>[0];
  * if the resolved module is a component it will include the project annotations
  */
 export const useOf = <TType extends ResolvedModuleExportType>(
-  moduleExportOrType: ModuleExport | TType,
+  moduleExportOrType: Of,
   validTypes?: TType[]
 ): ResolvedModuleExportFromType<TType> => {
   const context = useContext(DocsContext);