From 0f1eac5a98c14e63c8528a3e7ab11c61552ba6a6 Mon Sep 17 00:00:00 2001
From: Pavel Krymets <pavel@krymets.com>
Date: Fri, 28 Oct 2016 15:03:50 -0700
Subject: [PATCH] React to DependencyInjection changes (#869)

---
 .../HostingAbstractionsWebHostBuilderExtensions.cs             | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Microsoft.AspNetCore.Hosting.Abstractions/HostingAbstractionsWebHostBuilderExtensions.cs b/src/Microsoft.AspNetCore.Hosting.Abstractions/HostingAbstractionsWebHostBuilderExtensions.cs
index 767ed9263f..4b29b974d6 100644
--- a/src/Microsoft.AspNetCore.Hosting.Abstractions/HostingAbstractionsWebHostBuilderExtensions.cs
+++ b/src/Microsoft.AspNetCore.Hosting.Abstractions/HostingAbstractionsWebHostBuilderExtensions.cs
@@ -77,7 +77,8 @@ namespace Microsoft.AspNetCore.Hosting
             {
                 // It would be nicer if this was transient but we need to pass in the
                 // factory instance directly
-                services.AddSingleton(server);
+                // Registering as factory so server gets disposed along with a WebHost
+                services.AddSingleton(provider => server);
             });
         }