diff --git a/services/dialog_ui/enable_notification_dialog/entry/src/main/ets/Application/AbilityStage.ts b/services/dialog_ui/enable_notification_dialog/entry/src/main/ets/Application/AbilityStage.ts
index 1821f6288597deb3c74186db9ab9a7e3f4f21806..8f5ec9a9ec8902815cd90929fed5f0c589830628 100644
--- a/services/dialog_ui/enable_notification_dialog/entry/src/main/ets/Application/AbilityStage.ts
+++ b/services/dialog_ui/enable_notification_dialog/entry/src/main/ets/Application/AbilityStage.ts
@@ -13,7 +13,7 @@
  * limitations under the License.
  */
 
-import AbilityStage from "@ohos.application.AbilityStage"
+import AbilityStage from "@ohos.app.ability.AbilityStage"
 
 export default class DialogAbilityStage extends AbilityStage {
     onCreate() {
diff --git a/services/dialog_ui/enable_notification_dialog/entry/src/main/ets/ServiceExtAbility/NotificationServiceExtAbility.ts b/services/dialog_ui/enable_notification_dialog/entry/src/main/ets/ServiceExtAbility/NotificationServiceExtAbility.ts
index a7d0b9d1912ad3be1c094459211ff7aeab5c8ffe..122ae52931bdfd8eb3ccad6352e4d429b94d2155 100644
--- a/services/dialog_ui/enable_notification_dialog/entry/src/main/ets/ServiceExtAbility/NotificationServiceExtAbility.ts
+++ b/services/dialog_ui/enable_notification_dialog/entry/src/main/ets/ServiceExtAbility/NotificationServiceExtAbility.ts
@@ -13,7 +13,7 @@
  * limitations under the License.
  */
 
-import extension from '@ohos.application.ServiceExtensionAbility';
+import extension from '@ohos.app.ability.ServiceExtensionAbility';
 import window from '@ohos.window';
 import display from '@ohos.display';
 const TAG = "NotificationDialog_Service";