diff --git a/mock-generate/src/generate/generateMockJsFile.ts b/mock-generate/src/generate/generateMockJsFile.ts
index 660f0877..f10df62c 100644
--- a/mock-generate/src/generate/generateMockJsFile.ts
+++ b/mock-generate/src/generate/generateMockJsFile.ts
@@ -211,7 +211,7 @@ export function generateImportDeclaration(importEntity: ImportElementEntity, sou
  * @returns
  */
  function checIsDefaultExportClass(importName: string): boolean {
-  const defaultExportClass = ['Context', 'BaseContext', 'ExtensionContext', 'ApplicationContext', 'ExtensionAbility'];
+  const defaultExportClass = ['Context', 'BaseContext', 'ExtensionContext', 'ApplicationContext', 'ExtensionAbility', 'Ability'];
   return defaultExportClass.includes(importName);
 }