rls: fix flaky test introduced by #6514 (#6535)

grpc · Aug 10, 2023 · e274152 · e274152
1 parent 61a1f77
commit e274152
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/balancer/rls/balancer_test.go b/balancer/rls/balancer_test.go
@@ -919,7 +919,7 @@ func (s) TestUpdateStatePauses(t *testing.T) {
 	}
 	stub.Register(childPolicyName, stub.BalancerFuncs{
 		Init: func(bd *stub.BalancerData) {
-			bd.Data = balancer.Get(grpc.PickFirstBalancerName).Build(ccWrapper, bd.BuildOptions)
+			bd.Data = balancer.Get(grpc.PickFirstBalancerName).Build(bd.ClientConn, bd.BuildOptions)
 		},
 		ParseConfig: func(sc json.RawMessage) (serviceconfig.LoadBalancingConfig, error) {
 			cfg := &childPolicyConfig{}