diff --git a/entry/src/main/ets/common/utils/ReadFile.ets b/entry/src/main/ets/common/utils/ReadFile.ets
index 7998b100a4c41aaf757cc7eaec1fcef8e32f963e..5bfa86eb7571a4c892c8dab8dfec78216602c1dc 100644
--- a/entry/src/main/ets/common/utils/ReadFile.ets
+++ b/entry/src/main/ets/common/utils/ReadFile.ets
@@ -17,8 +17,10 @@ import { fileIo } from '@kit.CoreFileKit';
 import { common } from '@kit.AbilityKit';
 import { buffer } from '@kit.ArkTS';
 
+const uiContext: UIContext | undefined = AppStorage.get('uiContext');
+
 // Obtaining the Application File Path
-let context = getContext(this) as common.UIAbilityContext;
+let context = uiContext!.getHostContext() as common.UIAbilityContext;
 let filesDir = context.filesDir;
 let res: string = '';
 
diff --git a/entry/src/main/ets/common/utils/WriteFile.ets b/entry/src/main/ets/common/utils/WriteFile.ets
index 59a958c20562e4460b102521c1800075d5456564..b1cfd57e5f9f53638a0a87f71e16a809e8fb6007 100644
--- a/entry/src/main/ets/common/utils/WriteFile.ets
+++ b/entry/src/main/ets/common/utils/WriteFile.ets
@@ -16,7 +16,8 @@
 import { fileIo } from '@kit.CoreFileKit';
 import { common } from '@kit.AbilityKit';
 
-let context = getContext(this) as common.UIAbilityContext;
+const uiContext: UIContext | undefined = AppStorage.get('uiContext');
+let context = uiContext!.getHostContext() as common.UIAbilityContext;
 let filesDir = context.filesDir;
 
 /**
diff --git a/entry/src/main/ets/entryability/EntryAbility.ets b/entry/src/main/ets/entryability/EntryAbility.ets
index 43d55fab40beb98493dfd1a258c0264da3b107c7..9827b546ee821dd7c141a34e939b10a73e9cf803 100644
--- a/entry/src/main/ets/entryability/EntryAbility.ets
+++ b/entry/src/main/ets/entryability/EntryAbility.ets
@@ -36,6 +36,7 @@ export default class EntryAbility extends UIAbility {
         return;
       }
       hilog.info(0x0000, 'testTag', 'Succeeded in loading the content. Data: %{public}s', JSON.stringify(data) ?? '');
+      AppStorage.setOrCreate('uiContext', windowStage.getMainWindowSync().getUIContext());
     });
   }
 
diff --git a/entry/src/main/ets/view/PublicFilesTab.ets b/entry/src/main/ets/view/PublicFilesTab.ets
index ff9e90d96146b7f4816d73c5b48f60671a135c46..0f92e5ca69d40a15cc8775f2e9b7bd122de826e5 100644
--- a/entry/src/main/ets/view/PublicFilesTab.ets
+++ b/entry/src/main/ets/view/PublicFilesTab.ets
@@ -46,14 +46,14 @@ export struct publicFilesTab {
             if (result == SaveButtonOnClickResult.SUCCESS) {
               try {
                 Logger.info('createAsset successfully, event: ' + event);
-                let context = getContext();
+                let context = this.getUIContext().getHostContext();
                 let phAccessHelper = photoAccessHelper.getPhotoAccessHelper(context);
                 // Creating a Media File
                 let uri = await phAccessHelper.createAsset(photoAccessHelper.PhotoType.IMAGE, 'jpg');
                 Logger.info('createAsset successfully, uri: ' + uri);
                 // Open the created media file and read the local file and convert it to ArrayBuffer for easy filling.
                 let file = await fileIo.open(uri, fileIo.OpenMode.READ_WRITE);
-                let buffer = getContext(this).resourceManager.getMediaContentSync($r('app.media.img').id);
+                let buffer = this.getUIContext().getHostContext()!.resourceManager.getMediaContentSync($r('app.media.img').id);
                 // Write the read ArrayBuffer to the new media file.
                 let writeLen = await fileIo.write(file.fd, buffer.buffer);
                 Logger.info('write success,len=' + writeLen);