diff --git a/samples/AntiforgerySample/Startup.cs b/samples/AntiforgerySample/Startup.cs
index 8c00c95793..e2f548fc75 100644
--- a/samples/AntiforgerySample/Startup.cs
+++ b/samples/AntiforgerySample/Startup.cs
@@ -73,13 +73,13 @@ namespace AntiforgerySample
 
         public static void Main(string[] args)
         {
-            var application = new WebApplicationBuilder()
-                .UseConfiguration(WebApplicationConfiguration.GetDefault(args))
+            var host = new WebHostBuilder()
+                .UseDefaultConfiguration(args)
                 .UseIISPlatformHandlerUrl()
                 .UseStartup<Startup>()
                 .Build();
 
-            application.Run();
+            host.Run();
         }
     }
 }
diff --git a/test/Microsoft.AspNet.Antiforgery.FunctionalTests/AntiForgerySampleTestFixture.cs b/test/Microsoft.AspNet.Antiforgery.FunctionalTests/AntiForgerySampleTestFixture.cs
index 7e77232ed4..fbdb3e0ea9 100644
--- a/test/Microsoft.AspNet.Antiforgery.FunctionalTests/AntiForgerySampleTestFixture.cs
+++ b/test/Microsoft.AspNet.Antiforgery.FunctionalTests/AntiForgerySampleTestFixture.cs
@@ -23,7 +23,7 @@ namespace Microsoft.AspNet.Antiforgery.FunctionalTests
                 new KeyValuePair<string, string>("webroot", "wwwroot")
             });
 
-            var builder = new WebApplicationBuilder()
+            var builder = new WebHostBuilder()
                 .UseConfiguration(configurationBuilder.Build())
                 .UseStartup(typeof(AntiforgerySample.Startup))
                 .UseApplicationBasePath("../../../../samples/AntiforgerySample");