diff --git a/DatadogCore/Tests/Datadog/Core/Upload/DataUploadWorkerTests.swift b/DatadogCore/Tests/Datadog/Core/Upload/DataUploadWorkerTests.swift
index 2d10253231..d87f82975d 100644
--- a/DatadogCore/Tests/Datadog/Core/Upload/DataUploadWorkerTests.swift
+++ b/DatadogCore/Tests/Datadog/Core/Upload/DataUploadWorkerTests.swift
@@ -564,7 +564,7 @@ class DataUploadWorkerTests: XCTestCase {
         uploadExpectation.expectedFulfillmentCount = 3
 
         let dataUploader = DataUploaderMock(
-            uploadStatus: .mockRandom(),
+            uploadStatus: .mockWith(needsRetry: false),
             onUpload: uploadExpectation.fulfill
         )
         let worker = DataUploadWorker(