diff --git a/demo/ApiGateway/Program.cs b/demo/ApiGateway/Program.cs
index 2157126..029fd7b 100644
--- a/demo/ApiGateway/Program.cs
+++ b/demo/ApiGateway/Program.cs
@@ -1,6 +1,7 @@
 using Microsoft.AspNetCore;
 using Microsoft.AspNetCore.Hosting;
 using Microsoft.Extensions.Configuration;
+using Microsoft.Extensions.Hosting;
 using MMLib.SwaggerForOcelot.DependencyInjection;
 
 namespace ApiGateway
@@ -12,8 +13,8 @@ public static void Main(string[] args)
             CreateWebHostBuilder(args).Build().Run();
         }
 
-        public static IWebHostBuilder CreateWebHostBuilder(string[] args) =>
-            WebHost.CreateDefaultBuilder(args)
+        public static IHostBuilder CreateWebHostBuilder(string[] args) =>
+            Host.CreateDefaultBuilder(args)
                 .ConfigureAppConfiguration((hostingContext, config) =>
                 {
                     config
@@ -25,6 +26,6 @@ public static IWebHostBuilder CreateWebHostBuilder(string[] args) =>
                         .AddOcelotWithSwaggerSupport(folder: "Configuration")
                         .AddEnvironmentVariables();
                 })
-                .UseStartup<Startup>();
+                .ConfigureWebHostDefaults(webBuilder => webBuilder.UseStartup<Startup>());
     }
 }
diff --git a/src/MMLib.SwaggerForOcelot/DependencyInjection/ConfigurationBuilderExtensions.cs b/src/MMLib.SwaggerForOcelot/DependencyInjection/ConfigurationBuilderExtensions.cs
index 2cb99fd..2bf7673 100644
--- a/src/MMLib.SwaggerForOcelot/DependencyInjection/ConfigurationBuilderExtensions.cs
+++ b/src/MMLib.SwaggerForOcelot/DependencyInjection/ConfigurationBuilderExtensions.cs
@@ -1,6 +1,7 @@
 using Kros.Extensions;
 using Microsoft.AspNetCore.Hosting;
 using Microsoft.Extensions.Configuration;
+using Microsoft.Extensions.Hosting;
 using MMLib.SwaggerForOcelot.Configuration;
 using Newtonsoft.Json;
 using System;
@@ -28,7 +29,7 @@ public static class ConfigurationBuilderExtensions
         /// <returns>a Object IConfigurationBuilder</returns>
         public static IConfigurationBuilder AddOcelotWithSwaggerSupport(
             this IConfigurationBuilder builder,
-            IWebHostEnvironment environment = null,
+            IHostEnvironment environment = null,
             string folder = "/",
             string fileOfSwaggerEndPoints = SwaggerForOcelotFileOptions.SwaggerEndPointsConfigFile)
         {