diff --git a/dist/index.js b/dist/index.js
index d6b53bc..6ea37c7 100644
--- a/dist/index.js
+++ b/dist/index.js
@@ -8398,7 +8398,7 @@ function downloadRepository(accessToken, owner, repo, ref, commit, repositoryPat
                 assert.ok(archiveUrl, `Expected GitHub API response to contain 'Location' header`);
                 // Download the archive
                 core.info('Downloading the archive'); // Do not print the archive URL because it has an embedded token
-                downloadFile(archiveUrl, fileStream);
+                yield downloadFile(archiveUrl, fileStream);
             }
             finally {
                 yield fileStreamClosed;
diff --git a/src/github-api-helper.ts b/src/github-api-helper.ts
index e0f0cdc..8bec9a6 100644
--- a/src/github-api-helper.ts
+++ b/src/github-api-helper.ts
@@ -64,7 +64,7 @@ export async function downloadRepository(
 
       // Download the archive
       core.info('Downloading the archive') // Do not print the archive URL because it has an embedded token
-      downloadFile(archiveUrl, fileStream)
+      await downloadFile(archiveUrl, fileStream)
     } finally {
       await fileStreamClosed
     }