Fixed #70

unitycontainer · Dec 12, 2019 · ef0d013 · ef0d013
1 parent 8ea1ed7
commit ef0d013
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git 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
@@ -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)