From a66ef65fd9994d61a213ad8171f34e1891b35a02 Mon Sep 17 00:00:00 2001
From: Darren Shepherd <darren@rancher.com>
Date: Sat, 1 Feb 2020 15:58:24 -0700
Subject: [PATCH] Fix incorrect handler setup

---
 pkg/server/server.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pkg/server/server.go b/pkg/server/server.go
index 95c90d9..262355f 100644
--- a/pkg/server/server.go
+++ b/pkg/server/server.go
@@ -88,6 +88,8 @@ func (c *Server) Handler(ctx context.Context) (http.Handler, error) {
 		return nil, err
 	}
 
+	c.Next = handler
+
 	for _, hook := range c.StartHooks {
 		if err := hook(ctx, c); err != nil {
 			return nil, err
@@ -108,7 +110,7 @@ func (c *Server) Handler(ctx context.Context) (http.Handler, error) {
 		}
 	}
 
-	return handler, nil
+	return c.Next, nil
 }
 
 func (c *Server) ListenAndServe(ctx context.Context, httpsPort, httpPort int, opts *server.ListenOpts) error {