From de86071b2c8a21299b1a70139e31c3ca702c20a0 Mon Sep 17 00:00:00 2001
From: Javier Calvarro Nelson <jacalvar@microsoft.com>
Date: Thu, 22 Mar 2018 13:39:45 -0700
Subject: [PATCH] React to changes in Virtual schemes

---
 .../AzureAdB2CAuthenticationBuilderExtensions.cs       | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/Microsoft.AspNetCore.Authentication.AzureADB2C.UI/AzureAdB2CAuthenticationBuilderExtensions.cs b/src/Microsoft.AspNetCore.Authentication.AzureADB2C.UI/AzureAdB2CAuthenticationBuilderExtensions.cs
index a841e5f233..ab4b0240e8 100644
--- a/src/Microsoft.AspNetCore.Authentication.AzureADB2C.UI/AzureAdB2CAuthenticationBuilderExtensions.cs
+++ b/src/Microsoft.AspNetCore.Authentication.AzureADB2C.UI/AzureAdB2CAuthenticationBuilderExtensions.cs
@@ -51,9 +51,9 @@ namespace Microsoft.AspNetCore.Authentication
             string jwtBearerScheme,
             Action<AzureADB2COptions> configureOptions) {
 
-            builder.AddVirtualScheme(scheme, displayName: null, configureOptions: o =>
+            builder.AddPolicyScheme(scheme, displayName: null, configureOptions: o =>
             {
-                o.Default = jwtBearerScheme;
+                o.ForwardDefault = jwtBearerScheme;
             });
 
             builder.Services.Configure(TryAddJwtBearerSchemeMapping(scheme, jwtBearerScheme));
@@ -105,10 +105,10 @@ namespace Microsoft.AspNetCore.Authentication
             Action<AzureADB2COptions> configureOptions)
         {
             AddAdditionalMvcApplicationParts(builder.Services);
-            builder.AddVirtualScheme(scheme, displayName, o =>
+            builder.AddPolicyScheme(scheme, displayName, o =>
             {
-                o.Default = cookieScheme;
-                o.Challenge = openIdConnectScheme;
+                o.ForwardDefault = cookieScheme;
+                o.ForwardChallenge = openIdConnectScheme;
             });
 
             builder.Services.Configure(TryAddOpenIDCookieSchemeMappings(scheme, openIdConnectScheme, cookieScheme));