From ef0d01340ecf0e4fa4fb885af945a26394275dc5 Mon Sep 17 00:00:00 2001
From: Eugene Sadocoi <evgeni@eniks.com>
Date: Wed, 11 Dec 2019 22:47:43 -0800
Subject: [PATCH] Fixed #70

---
 package.props                      |  2 +-
 src/Extensions/HostingExtension.cs | 11 ++++++-----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/package.props b/package.props
index 2f31c1d..7aaa6f1 100644
--- a/package.props
+++ b/package.props
@@ -1,7 +1,7 @@
 <Project>
 
   <PropertyGroup>
-    <Version>5.11.4</Version>
+    <Version>5.11.5</Version>
     <PackageReleaseNotes>This release is compatible with Core 3.1</PackageReleaseNotes>
   </PropertyGroup>
 
diff --git a/src/Extensions/HostingExtension.cs b/src/Extensions/HostingExtension.cs
index 2e278bb..a1178f3 100644
--- a/src/Extensions/HostingExtension.cs
+++ b/src/Extensions/HostingExtension.cs
@@ -14,11 +14,12 @@ public static IHostBuilder UseUnityServiceProvider(this IHostBuilder hostBuilder
         {
             _factory = new ServiceProviderFactory(container);
 
-            return hostBuilder.ConfigureServices((context, services) =>
-            {
-                services.Replace(ServiceDescriptor.Singleton<IServiceProviderFactory<IUnityContainer>>(_factory));
-                services.Replace(ServiceDescriptor.Singleton<IServiceProviderFactory<IServiceCollection>>(_factory));
-            });
+            return hostBuilder.UseServiceProviderFactory<IUnityContainer>(_factory)
+                              .ConfigureServices((context, services) =>
+                              {
+                                  services.Replace(ServiceDescriptor.Singleton<IServiceProviderFactory<IUnityContainer>>(_factory));
+                                  services.Replace(ServiceDescriptor.Singleton<IServiceProviderFactory<IServiceCollection>>(_factory));
+                              });
         }
 
         public static IWebHostBuilder UseUnityServiceProvider(this IWebHostBuilder hostBuilder, IUnityContainer container = null)