From cf580771485dd832be3aed2cc8b0a95a1160ad6e Mon Sep 17 00:00:00 2001
From: Hao Kung <haok@microsoft.com>
Date: Tue, 23 Jun 2015 17:40:02 -0700
Subject: [PATCH] React to Http changes

---
 src/Microsoft.AspNet.Session/SessionMiddleware.cs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/Microsoft.AspNet.Session/SessionMiddleware.cs b/src/Microsoft.AspNet.Session/SessionMiddleware.cs
index 57c1d8f..2735aa0 100644
--- a/src/Microsoft.AspNet.Session/SessionMiddleware.cs
+++ b/src/Microsoft.AspNet.Session/SessionMiddleware.cs
@@ -92,16 +92,17 @@ public SessionEstablisher(HttpContext context, string sessionKey, SessionOptions
                 _context = context;
                 _sessionKey = sessionKey;
                 _options = options;
-                context.Response.OnResponseStarting(OnResponseStartingCallback, state: this);
+                context.Response.OnStarting(OnStartingCallback, state: this);
             }
 
-            private static void OnResponseStartingCallback(object state)
+            private static Task OnStartingCallback(object state)
             {
                 var establisher = (SessionEstablisher)state;
                 if (establisher._shouldEstablishSession)
                 {
                     establisher.SetCookie();
                 }
+                return Task.FromResult(0);
             }
 
             private void SetCookie()