diff --git a/backend/internal/service/widget_service_test.go b/backend/internal/service/widget_service_test.go
index 04e634b4..3ed192f1 100644
--- a/backend/internal/service/widget_service_test.go
+++ b/backend/internal/service/widget_service_test.go
@@ -311,6 +311,9 @@ func TestWidgetService_PublicCreateMessageRollsBackCaptainJobFailureAndRetries(t
 	message, _, attachments, err := svc.PublicCreateMessage(context.Background(), "public-api", "visitor-source", displayID, req)
 	require.NoError(t, err)
 	require.Len(t, attachments, 1)
+	var retriedUpload model.DirectUpload
+	require.NoError(t, db.First(&retriedUpload, upload.ID).Error)
+	assert.Equal(t, model.DirectUploadStatusCompleted, retriedUpload.Status)
 	assert.Len(t, listener.events, 1)
 	var job model.BackgroundJob
 	require.NoError(t, db.Where("job_type = ?", TaskTypeCaptainConversationResponseBuilder).First(&job).Error)