diff --git a/mock-generate/src/generate/generateCommonFunction.ts b/mock-generate/src/generate/generateCommonFunction.ts
index 9185c17f..8c337a18 100644
--- a/mock-generate/src/generate/generateCommonFunction.ts
+++ b/mock-generate/src/generate/generateCommonFunction.ts
@@ -39,6 +39,8 @@ export function generateCommonFunction(rootName: string, functionArray: Array<Fu
     if (functionEntity.returnType.returnKind !== SyntaxKind.VoidKeyword) {
       if (rootName === 'featureAbility' && functionEntity.returnType.returnKindName === 'Context') {
         functionBody += 'return _Context;';
+      } else if (rootName === 'inputMethod' && functionEntity.returnType.returnKindName === 'InputMethodSubtype') {
+        functionBody += 'return mockInputMethodSubtype().InputMethodSubtype;'
       } else {
         functionBody += getReturnStatement(functionEntity.returnType, sourceFile);
       }
diff --git a/mock-generate/src/generate/generateMockJsFile.ts b/mock-generate/src/generate/generateMockJsFile.ts
index f3775f59..660f0877 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'];
+  const defaultExportClass = ['Context', 'BaseContext', 'ExtensionContext', 'ApplicationContext', 'ExtensionAbility'];
   return defaultExportClass.includes(importName);
 }