diff --git a/src/types.ts b/src/types.ts
index 3732a16582..162e03e7ff 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -7,8 +7,9 @@ import type { IModify } from '@umijs/core';
 import type { AssetsPackage, ExampleBlockAsset } from 'dumi-assets-types';
 import type { Element } from 'hast';
 import type { IApi as IUmiApi } from 'umi';
+import { defineConfig as defineUmiConfig } from 'umi';
 
-type IUmiConfig = IUmiApi['config'];
+type IUmiConfig = Parameters<typeof defineUmiConfig>[0];
 
 export interface IDumiConfig extends IUmiConfig {
   resolve: {