diff --git a/library/ui/src/main/java/com/google/android/exoplayer2/ui/DownloadNotificationHelper.java b/library/ui/src/main/java/com/google/android/exoplayer2/ui/DownloadNotificationHelper.java
index 775dea253f..278cfeca59 100644
--- a/library/ui/src/main/java/com/google/android/exoplayer2/ui/DownloadNotificationHelper.java
+++ b/library/ui/src/main/java/com/google/android/exoplayer2/ui/DownloadNotificationHelper.java
@@ -64,13 +64,12 @@ public final class DownloadNotificationHelper {
     boolean haveDownloadTasks = false;
     boolean haveRemoveTasks = false;
     for (Download download : downloads) {
-      if (download.state == Download.STATE_REMOVING
-          || download.state == Download.STATE_RESTARTING) {
+      if (download.state == Download.STATE_REMOVING) {
         haveRemoveTasks = true;
         continue;
       }
-      if (download.state != Download.STATE_DOWNLOADING
-          && download.state != Download.STATE_COMPLETED) {
+      if (download.state != Download.STATE_RESTARTING
+          && download.state != Download.STATE_DOWNLOADING) {
         continue;
       }
       haveDownloadTasks = true;