diff --git a/webprerenderlibrary/src/main/ets/common/utils/WindowUtil.ets b/webprerenderlibrary/src/main/ets/common/utils/WindowUtil.ets
index ab3154131010bc51d5ea2cd304ad09c135f7de74..5ee2b40fc008b46a8d197094b2a3c2086266624d 100644
--- a/webprerenderlibrary/src/main/ets/common/utils/WindowUtil.ets
+++ b/webprerenderlibrary/src/main/ets/common/utils/WindowUtil.ets
@@ -76,11 +76,11 @@ export class WindowUtil {
 
   // Get status bar height and indicator height.
   private static setAvoidArea(data: window.Window, type: window.AvoidAreaType, area: window.AvoidArea) {
-    const context: UIContext = data.getUIContext();
+    const context: UIContext | undefined = AppStorage.get<UIContext>(Constants.KEY_UI_CONTEXT);
     if (type === window.AvoidAreaType.TYPE_SYSTEM) {
-      AppStorage.setOrCreate(Constants.KEY_STATUS_BAR_HEIGHT, context.px2vp(area.topRect.height));
+      AppStorage.setOrCreate(Constants.KEY_STATUS_BAR_HEIGHT, context?.px2vp(area.topRect.height));
     } else if (type === window.AvoidAreaType.TYPE_NAVIGATION_INDICATOR) {
-      AppStorage.setOrCreate(Constants.KEY_NAV_INDICATOR_HEIGHT, context.px2vp(area.bottomRect.height));
+      AppStorage.setOrCreate(Constants.KEY_NAV_INDICATOR_HEIGHT, context?.px2vp(area.bottomRect.height));
     }
   }
 }
\ No newline at end of file
diff --git a/webprerendersample/src/main/ets/entryability/EntryAbility.ets b/webprerendersample/src/main/ets/entryability/EntryAbility.ets
index e7d674cb7af40acacaed2f6093f448f72e446192..7f65db3e6388608765a9abe10343943dab4e218b 100644
--- a/webprerendersample/src/main/ets/entryability/EntryAbility.ets
+++ b/webprerendersample/src/main/ets/entryability/EntryAbility.ets
@@ -15,44 +15,52 @@
 
 import { ConfigurationConstant, UIAbility } from '@kit.AbilityKit';
 import { window } from '@kit.ArkUI';
+import { BusinessError } from '@kit.BasicServicesKit';
 import { hilog } from '@kit.PerformanceAnalysisKit';
 import { WebPrerenderController } from '@ohos_samples/webprerenderlibrary';
 
+const TAG = '[WebPrerenderSampleAbility]';
+
 export default class WebPrerenderSampleAbility extends UIAbility {
   onCreate(): void {
-    hilog.info(0x0000, 'testTag', '%{public}s', 'Ability onCreate');
-    this.context.getApplicationContext().setColorMode(ConfigurationConstant.ColorMode.COLOR_MODE_NOT_SET);
+    hilog.info(0x0000, TAG, '%{public}s', 'Ability onCreate');
+    try {
+      this.context.getApplicationContext().setColorMode(ConfigurationConstant.ColorMode.COLOR_MODE_NOT_SET);
+    } catch (error) {
+      const err: BusinessError = error as BusinessError;
+      hilog.error(0x0000, TAG, '%{public}s', `Set colorMode failed: ${err.code}, ${err.message}`);
+    }
   }
 
   onDestroy(): void {
-    hilog.info(0x0000, 'testTag', '%{public}s', 'Ability onDestroy');
+    hilog.info(0x0000, TAG, '%{public}s', 'Ability onDestroy');
   }
 
   onWindowStageCreate(windowStage: window.WindowStage): void {
     // Main window is created, set main page for this ability
-    hilog.info(0x0000, 'testTag', '%{public}s', 'Ability onWindowStageCreate');
+    hilog.info(0x0000, TAG, '%{public}s', 'Ability onWindowStageCreate');
     windowStage.loadContent('pages/Index', (err) => {
       if (err.code) {
-        hilog.error(0x0000, 'testTag', '%{public}s', `Failed to load the content. Cause: ${err.code}`);
+        hilog.error(0x0000, TAG, '%{public}s', `Failed to load the content. Cause: ${err.code}`);
         return;
       }
       WebPrerenderController.initWindowConfig(windowStage);
-      hilog.info(0x0000, 'testTag', '%{public}s', `Succeed in loading the content.`);
+      hilog.info(0x0000, TAG, '%{public}s', `Succeed in loading the content.`);
     });
   }
 
   onWindowStageDestroy(): void {
     // Main window is destroyed, release UI related resources
-    hilog.info(0x0000, 'testTag', '%{public}s', 'Ability onWindowStageDestroy');
+    hilog.info(0x0000, TAG, '%{public}s', 'Ability onWindowStageDestroy');
   }
 
   onForeground(): void {
     // Ability has brought to foreground
-    hilog.info(0x0000, 'testTag', '%{public}s', 'Ability onForeground');
+    hilog.info(0x0000, TAG, '%{public}s', 'Ability onForeground');
   }
 
   onBackground(): void {
     // Ability has back to background
-    hilog.info(0x0000, 'testTag', '%{public}s', 'Ability onBackground');
+    hilog.info(0x0000, TAG, '%{public}s', 'Ability onBackground');
   }
 }
\ No newline at end of file