From 526e5c8032ad4e27d8b03b04f27fc8960941f867 Mon Sep 17 00:00:00 2001
From: Arthur Vickers <ajcvickers@hotmail.com>
Date: Fri, 11 Mar 2016 10:34:06 -0800
Subject: [PATCH] React to EF D.I. changes

---
 .../TestHelperExtensions.cs                                   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/Microsoft.AspNetCore.Diagnostics.EntityFrameworkCore.Tests/TestHelperExtensions.cs b/test/Microsoft.AspNetCore.Diagnostics.EntityFrameworkCore.Tests/TestHelperExtensions.cs
index d101ba3665..e87b4ef99d 100644
--- a/test/Microsoft.AspNetCore.Diagnostics.EntityFrameworkCore.Tests/TestHelperExtensions.cs
+++ b/test/Microsoft.AspNetCore.Diagnostics.EntityFrameworkCore.Tests/TestHelperExtensions.cs
@@ -8,9 +8,9 @@ namespace Microsoft.EntityFrameworkCore.Tests
 {
     public static class TestHelperExtensions
     {
-        public static EntityFrameworkServicesBuilder AddProviderServices(this EntityFrameworkServicesBuilder entityServicesBuilder)
+        public static IServiceCollection AddProviderServices(this IServiceCollection serviceCollection)
         {
-            return entityServicesBuilder.AddInMemoryDatabase();
+            return serviceCollection.AddEntityFrameworkInMemoryDatabase();
         }
 
         public static DbContextOptions UseProviderOptions(this DbContextOptions options)