diff --git a/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/logging/BeamFnLoggingClientTest.java b/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/logging/BeamFnLoggingClientTest.java
index 0ba56047d0c3..2dd2b2bad53f 100644
--- a/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/logging/BeamFnLoggingClientTest.java
+++ b/sdks/java/harness/src/test/java/org/apache/beam/fn/harness/logging/BeamFnLoggingClientTest.java
@@ -532,6 +532,7 @@ public StreamObserver<BeamFnApi.LogEntry.List> logging(
 
     ManagedChannel channel = InProcessChannelBuilder.forName(apiServiceDescriptor.getUrl()).build();
     try {
+      thrown.expectMessage("Logging stream terminated unexpectedly");
       BeamFnLoggingClient client =
           BeamFnLoggingClient.createAndStart(
               PipelineOptionsFactory.fromArgs(
@@ -543,7 +544,6 @@ public StreamObserver<BeamFnApi.LogEntry.List> logging(
               apiServiceDescriptor,
               (Endpoints.ApiServiceDescriptor descriptor) -> channel);
 
-      thrown.expectMessage("Logging stream terminated unexpectedly");
       client.terminationFuture().get();
     } finally {
       // Verify that after termination, log levels are reset.