diff --git a/app/src/main/java/org/lineageos/updater/UpdaterReceiver.java b/app/src/main/java/org/lineageos/updater/UpdaterReceiver.java
index f412f4e3295b2f24f19fd428c51b470d24d30cc2..c8857db00b933d74ddf6c8a301b18cef2005e974 100644
--- a/app/src/main/java/org/lineageos/updater/UpdaterReceiver.java
+++ b/app/src/main/java/org/lineageos/updater/UpdaterReceiver.java
@@ -28,10 +28,10 @@ import android.os.SystemProperties;
 import androidx.core.app.NotificationCompat;
 import androidx.preference.PreferenceManager;
 
-import org.lineageos.updater.controller.UpdaterController;
 import org.lineageos.updater.misc.BuildInfoUtils;
 import org.lineageos.updater.misc.Constants;
 import org.lineageos.updater.misc.StringGenerator;
+import org.lineageos.updater.misc.Utils;
 
 import java.text.DateFormat;
 
@@ -87,18 +87,14 @@ public class UpdaterReceiver extends BroadcastReceiver {
 
     @Override
     public void onReceive(Context context, Intent intent) {
-        SharedPreferences pref = PreferenceManager.getDefaultSharedPreferences(context);
         if (ACTION_INSTALL_REBOOT.equals(intent.getAction())) {
-            String downloadId = pref.getString(Constants.PREF_NEEDS_DELETE_ID, null);
-            UpdaterController controller = UpdaterController.getInstance(context);
-            if (controller != null && downloadId != null) {
-                controller.deleteUpdate(downloadId);
-            }
+            Utils.removeInstalledUpdate(context);
             PowerManager pm = context.getSystemService(PowerManager.class);
             pm.reboot(null);
         } else if (Intent.ACTION_BOOT_COMPLETED.equals(intent.getAction())) {
+            SharedPreferences pref = PreferenceManager.getDefaultSharedPreferences(context);
             pref.edit().remove(Constants.PREF_NEEDS_REBOOT_ID).apply();
-            pref.edit().remove(Constants.PREF_NEEDS_DELETE_ID).apply();
+            Utils.removeInstalledUpdate(context);
 
             if (shouldShowUpdateFailedNotification(context)) {
                 pref.edit().putBoolean(Constants.PREF_INSTALL_NOTIFIED, true).apply();
diff --git a/app/src/main/java/org/lineageos/updater/misc/Utils.java b/app/src/main/java/org/lineageos/updater/misc/Utils.java
index 43b276d1a4afc1dc7244f817e3bad0df4e57673a..f1c9dac43ea5ee3827b443b3d0c0f51717045d6e 100644
--- a/app/src/main/java/org/lineageos/updater/misc/Utils.java
+++ b/app/src/main/java/org/lineageos/updater/misc/Utils.java
@@ -54,6 +54,7 @@ import org.json.JSONException;
 import org.json.JSONObject;
 import org.lineageos.updater.R;
 import org.lineageos.updater.UpdatesDbHelper;
+import org.lineageos.updater.controller.UpdaterController;
 import org.lineageos.updater.controller.UpdaterService;
 import org.lineageos.updater.model.Update;
 import org.lineageos.updater.model.UpdateBaseInfo;
@@ -484,6 +485,16 @@ public class Utils {
         preferences.edit().putBoolean(DOWNLOADS_CLEANUP_DONE, true).apply();
     }
 
+    public static void removeInstalledUpdate(Context context) {
+        SharedPreferences pref = PreferenceManager.getDefaultSharedPreferences(context);
+        String downloadId = pref.getString(Constants.PREF_NEEDS_DELETE_ID, null);
+        UpdaterController controller = UpdaterController.getInstance(context);
+        if (controller != null && downloadId != null) {
+            controller.deleteUpdate(downloadId);
+            pref.edit().remove(Constants.PREF_NEEDS_DELETE_ID).apply();
+        }
+    }
+
     public static File appendSequentialNumber(final File file) {
         String name;
         String extension;