fixup connected status

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

kgo.go

@@ -142,11 +142,9 @@ func (k *Broker) connect(ctx context.Context, opts ...kgo.Opt) (*kgo.Client, *ho
 			}
 			return nil, nil, err
 		}
+		k.connected.Store(1)
+		return c, htracer, nil
 	}
-
-	k.connected.Store(1)
-
-	return c, htracer, nil
 }
 
 func (k *Broker) Connect(ctx context.Context) error {