2 files changed, 1 insertions, 2 deletions

diff --git a/filedistribution/src/main/java/com/yahoo/vespa/filedistribution/Downloads.java b/filedistribution/src/main/java/com/yahoo/vespa/filedistribution/Downloads.java
index d364e9ec48d..181c424ef98 100644
--- a/filedistribution/src/main/java/com/yahoo/vespa/filedistribution/Downloads.java
+++ b/filedistribution/src/main/java/com/yahoo/vespa/filedistribution/Downloads.java
@@ -38,8 +38,8 @@ public class Downloads {
 
     void completedDownloading(FileReference fileReference, File file) {
         Optional<FileReferenceDownload> download = get(fileReference);
+        setDownloadStatus(fileReference, 1.0);
         if (download.isPresent()) {
-            downloadStatuses().get(fileReference).ifPresent(Downloads.DownloadStatus::finished);
             downloads.remove(fileReference);
             download.get().future().complete(Optional.of(file));
         } else {
diff --git a/filedistribution/src/test/java/com/yahoo/vespa/filedistribution/FileDownloaderTest.java b/filedistribution/src/test/java/com/yahoo/vespa/filedistribution/FileDownloaderTest.java
index 287cc7a74f3..475228543eb 100644
--- a/filedistribution/src/test/java/com/yahoo/vespa/filedistribution/FileDownloaderTest.java
+++ b/filedistribution/src/test/java/com/yahoo/vespa/filedistribution/FileDownloaderTest.java
@@ -122,7 +122,6 @@ public class FileDownloaderTest {
 
             // Verify download status when downloaded
             System.out.println(downloads.downloadStatuses());
-            double downloadStatus = downloads.downloadStatus(fileReference);
             assertDownloadStatus(fileReference, 1.0);
         }