diff --git a/app-common/src/main/java/im/angry/openeuicc/ui/wizard/DownloadWizardActivity.kt b/app-common/src/main/java/im/angry/openeuicc/ui/wizard/DownloadWizardActivity.kt
index 7cb7168..a9f868f 100644
--- a/app-common/src/main/java/im/angry/openeuicc/ui/wizard/DownloadWizardActivity.kt
+++ b/app-common/src/main/java/im/angry/openeuicc/ui/wizard/DownloadWizardActivity.kt
@@ -3,7 +3,6 @@ package im.angry.openeuicc.ui.wizard
 import android.app.assist.AssistContent
 import android.os.Bundle
 import android.view.View
-import android.view.WindowManager
 import android.view.inputmethod.InputMethodManager
 import android.widget.Button
 import android.widget.ProgressBar
@@ -317,14 +316,4 @@ class DownloadWizardActivity: BaseEuiccAccessActivity() {
 
         open fun beforeNext() {}
     }
-
-    override fun onStart() {
-        super.onStart()
-        window.addFlags(WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON)
-    }
-
-    override fun onStop() {
-        super.onStop()
-        window.clearFlags(WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON)
-    }
 }
\ No newline at end of file
diff --git a/app-common/src/main/res/layout/fragment_download_progress.xml b/app-common/src/main/res/layout/fragment_download_progress.xml
index 82ebb25..9d6b293 100644
--- a/app-common/src/main/res/layout/fragment_download_progress.xml
+++ b/app-common/src/main/res/layout/fragment_download_progress.xml
@@ -24,6 +24,7 @@
         android:id="@+id/download_progress_list"
         android:layout_width="0dp"
         android:layout_height="wrap_content"
+        android:keepScreenOn="true"
         app:layout_constraintTop_toBottomOf="@id/download_progress_title"
         app:layout_constraintStart_toStartOf="parent"
         app:layout_constraintEnd_toEndOf="parent"