diff --git a/lnd.go b/lnd.go index 13c4974a8..f8377f335 100644 --- a/lnd.go +++ b/lnd.go @@ -901,14 +901,7 @@ func startRestProxy(cfg *Config, rpcServer *rpcServer, restDialOpts []grpc.DialO ) // Register our services with the REST proxy. - err := lnrpc.RegisterStateHandlerFromEndpoint( - ctx, mux, restProxyDest, restDialOpts, - ) - if err != nil { - return nil, err - } - - err = rpcServer.RegisterWithRestProxy( + err := rpcServer.RegisterWithRestProxy( ctx, mux, restDialOpts, restProxyDest, ) if err != nil { diff --git a/rpcserver.go b/rpcserver.go index af127c648..2a1bacfeb 100644 --- a/rpcserver.go +++ b/rpcserver.go @@ -906,6 +906,15 @@ func (r *rpcServer) RegisterWithRestProxy(restCtx context.Context, return err } + // Register our State service with the REST proxy. + err = lnrpc.RegisterStateHandlerFromEndpoint( + restCtx, restMux, restProxyDest, restDialOpts, + ) + if err != nil { + return err + } + + // Register all the subservers with the REST proxy. for _, subServer := range r.subGrpcHandlers { err := subServer.RegisterWithRestServer( restCtx, restMux, restProxyDest, restDialOpts,