diff --git a/SJSAssetExportSession/SampleWriter.swift b/SJSAssetExportSession/SampleWriter.swift
index 90b88ce..6922cdd 100644
--- a/SJSAssetExportSession/SampleWriter.swift
+++ b/SJSAssetExportSession/SampleWriter.swift
@@ -131,10 +131,15 @@ actor SampleWriter {
         }
 
         progressContinuation?.yield(1.0)
-        progressContinuation?.finish()
 
         // Make sure the last progress value is yielded before returning.
         await Task.yield()
+        await withCheckedContinuation { continuation in
+            progressContinuation?.onTermination = { _ in
+                continuation.resume(returning: ())
+            }
+            progressContinuation?.finish()
+        }
     }
 
     // MARK: - Setup