diff --git a/clientv3/integration/server_shutdown_test.go b/clientv3/integration/server_shutdown_test.go
index 23115e0e610..ddd49845c5c 100644
--- a/clientv3/integration/server_shutdown_test.go
+++ b/clientv3/integration/server_shutdown_test.go
@@ -48,7 +48,12 @@ func TestBalancerUnderServerShutdownWatch(t *testing.T) {
 	defer watchCli.Close()
 
 	// wait for eps[lead] to be pinned
-	watchCli.Get(context.Background(), "foo")
+	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
+	_, err = watchCli.Get(ctx, "foo")
+	cancel()
+	if err != nil {
+		t.Fatal(err)
+	}
 
 	// add all eps to list, so that when the original pined one fails
 	// the client can switch to other available eps