From ecc5874b6c92af8a8bcea19344828605c774920b Mon Sep 17 00:00:00 2001
From: Chris R <Tratcher@outlook.com>
Date: Mon, 31 Aug 2015 10:51:48 -0700
Subject: [PATCH] Fix test.

---
 .../ElmMiddlewareTest.cs                                   | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/test/Microsoft.AspNet.Diagnostics.Tests/ElmMiddlewareTest.cs b/test/Microsoft.AspNet.Diagnostics.Tests/ElmMiddlewareTest.cs
index 11ca1f51c6..3da25074b4 100644
--- a/test/Microsoft.AspNet.Diagnostics.Tests/ElmMiddlewareTest.cs
+++ b/test/Microsoft.AspNet.Diagnostics.Tests/ElmMiddlewareTest.cs
@@ -219,8 +219,11 @@ namespace Microsoft.AspNet.Diagnostics.Tests
                 .Returns(true);
             var requestIdentifier = new Mock<IHttpRequestIdentifierFeature>();
             requestIdentifier.Setup(f => f.TraceIdentifier).Returns(Guid.NewGuid().ToString());
-            contextMock.Setup(c => c.Features.Get<IHttpRequestIdentifierFeature>())
-                .Returns(requestIdentifier.Object);
+            var featureCollection = new FeatureCollection();
+            featureCollection.Set<IHttpRequestIdentifierFeature>(requestIdentifier.Object);
+            contextMock
+                .SetupGet(c => c.Features)
+                .Returns(featureCollection);
             return contextMock;
         }
 #endif