diff --git a/entry/src/main/ets/entryability/EntryAbility.ts b/entry/src/main/ets/entryability/EntryAbility.ts
index 6d0226717e01a87efc562299818fb65243cdd3c9..45906183c344852fc01046f485610623454fb949 100644
--- a/entry/src/main/ets/entryability/EntryAbility.ts
+++ b/entry/src/main/ets/entryability/EntryAbility.ts
@@ -43,6 +43,7 @@ export default class EntryAbility extends UIAbility {
       }
       hilog.isLoggable(0x0000, 'testTag', hilog.LogLevel.INFO);
       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/model/PreferenceModel.ets b/entry/src/main/ets/model/PreferenceModel.ets
index c737f4338b8f573f59a6f2f4d0e2930b92eaef3c..c75c5cbc3a07eaedb3d144c51ad45c4743749afc 100644
--- a/entry/src/main/ets/model/PreferenceModel.ets
+++ b/entry/src/main/ets/model/PreferenceModel.ets
@@ -19,7 +19,8 @@ import Logger from '../common/utils/Logger';
 import CommonConstants from '../common/constants/CommonConstants';
 import Fruit from '../viewmodel/Fruit';
 
-let context = getContext(this);
+const uiContext: UIContext | undefined = AppStorage.get('uiContext');
+let context = uiContext!.getHostContext()!;
 let preference: preferences.Preferences;
 let preferenceTemp: preferences.Preferences;
 
@@ -140,7 +141,7 @@ class PreferenceModel {
    * @param message Prompt message.
    */
   showToastMessage(message: Resource) {
-    promptAction.showToast({
+    uiContext!.getPromptAction().showToast({
       message: message,
       duration: CommonConstants.DURATION
     });