diff --git a/samples/RoutingSample.Web/Startup.cs b/samples/RoutingSample.Web/Startup.cs index 1cc7d790c3..bcedf27a65 100644 --- a/samples/RoutingSample.Web/Startup.cs +++ b/samples/RoutingSample.Web/Startup.cs @@ -8,7 +8,8 @@ namespace RoutingSample.Web { public void Configuration(IBuilder builder) { - var routes = builder.UseRouter(); + var routes = new RouteCollection(); + builder.UseRouter(routes); var endpoint1 = new DelegateRouteEndpoint(async (context) => await context.HttpContext.Response.WriteAsync( diff --git a/src/Microsoft.AspNet.Routing/BuilderExtensions.cs b/src/Microsoft.AspNet.Routing/BuilderExtensions.cs index 7abc325f3c..38ec9914da 100644 --- a/src/Microsoft.AspNet.Routing/BuilderExtensions.cs +++ b/src/Microsoft.AspNet.Routing/BuilderExtensions.cs @@ -6,15 +6,10 @@ namespace Microsoft.AspNet.Abstractions { public static class BuilderExtensions { - public static IRouteCollection UseRouter(this IBuilder builder) - { - return UseRouter(builder, new RouteCollection()); - } - - public static IRouteCollection UseRouter(this IBuilder builder, IRouteCollection routes) + public static IBuilder UseRouter(this IBuilder builder, IRouteCollection routes) { builder.Use((next) => new RouterMiddleware(next, routes).Invoke); - return routes; + return builder; } } }