Revert unnecessary change

Author: arjan-bal

balancer/weightedtarget/weightedtarget_test.go

@@ -298,6 +298,7 @@ func (s) TestWeightedTarget(t *testing.T) {
 	}
 	verifyAddressInNewSubConn(t, cc, addr3)
 
+	// The subconn from the test_config_balancer should be shut down.
 	scShutdown = <-cc.ShutdownSubConnCh
 	// The same SubConn is closed by gracefulswitch and pickfirstleaf when they
 	// are closed. Remove duplicate events.
@@ -306,7 +307,6 @@ func (s) TestWeightedTarget(t *testing.T) {
 	// shutting down SubConns.
 	<-cc.ShutdownSubConnCh
 
-	// The subconn from the test_config_balancer should be shut down.
 	if scShutdown != sc2 {
 		t.Fatalf("ShutdownSubConn, want %v, got %v", sc2, scShutdown)
 	}

internal/balancergroup/balancergroup_test.go

@@ -360,8 +360,8 @@ func (s) TestBalancerGroup_locality_caching_not_read_within_timeout(t *testing.T
 	ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
 	defer cancel()
 	scToShutdown := map[balancer.SubConn]int{
-		addrToSC[testBackendAddrs[2].Addr]: 3,
-		addrToSC[testBackendAddrs[3].Addr]: 3,
+		addrToSC[testBackendAddrs[2].Addr]: 1,
+		addrToSC[testBackendAddrs[3].Addr]: 1,
 	}
 	for i := 0; i < len(scToShutdown); i++ {
 		select {