diff --git a/pkg/common/discoveryregister/direct/directconn.go b/pkg/common/discoveryregister/direct/directconn.go
index 3a455b2e9..c045764fb 100644
--- a/pkg/common/discoveryregister/direct/directconn.go
+++ b/pkg/common/discoveryregister/direct/directconn.go
@@ -95,7 +95,7 @@ func (cm *ConnManager) GetConn(ctx context.Context, serviceName string, opts ...
 		return nil, errs.Wrap(errors.New("unknown service name"), "serviceName", serviceName)
 	}
 
-	conn, err := dialService(address, append(cm.additionalOpts, opts...)...)
+	conn, err := dialService(ctx, address, append(cm.additionalOpts, opts...)...)
 	if err != nil {
 		return nil, err
 	}
@@ -117,9 +117,9 @@ func (cm *ConnManager) CloseConn(conn *grpc.ClientConn) {
 	}
 }
 
-func dialService(address string, opts ...grpc.DialOption) (*grpc.ClientConn, error) {
+func dialService(ctx context.Context, address string, opts ...grpc.DialOption) (*grpc.ClientConn, error) {
 	options := append(opts, grpc.WithInsecure()) // Replace WithInsecure with proper security options
-	conn, err := grpc.DialContext(context.Background(), address, options...)
+	conn, err := grpc.DialContext(ctx, address, options...)
 	if err != nil {
 		return nil, err
 	}