diff --git a/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/ConnectionWorkerPoolTest.java b/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/ConnectionWorkerPoolTest.java
index ccae51dcbd..4dc2f693e9 100644
--- a/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/ConnectionWorkerPoolTest.java
+++ b/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/ConnectionWorkerPoolTest.java
@@ -55,7 +55,7 @@ public class ConnectionWorkerPoolTest {
   private static MockServiceHelper serviceHelper;
   private BigQueryWriteSettings clientSettings;
 
-  private static final String TEST_TRACE_ID = "home:job1";
+  private static final String TEST_TRACE_ID = "DATAFLOW:job_id";
   private static final String TEST_STREAM_1 = "projects/p1/datasets/d1/tables/t1/streams/_default";
   private static final String TEST_STREAM_2 = "projects/p1/datasets/d1/tables/t2/streams/_default";
 
@@ -417,6 +417,9 @@ public void testCloseExternalClient()
       assertThat(response.getAppendResult().getOffset().getValue()).isEqualTo(i);
     }
     assertThat(testBigQueryWrite.getAppendRequests().size()).isEqualTo(appendCount * 2);
+    for (int i = 0; i < streamWriterList.size(); i++) {
+      streamWriterList.get(i).close();
+    }
   }
 
   private AppendRowsResponse createAppendResponse(long offset) {