diff --git a/test/Microsoft.AspNet.Diagnostics.Entity.Tests/TestHelperExtensions.cs b/test/Microsoft.AspNet.Diagnostics.Entity.Tests/TestHelperExtensions.cs
index d9e793e6e6..27ca02fc22 100644
--- a/test/Microsoft.AspNet.Diagnostics.Entity.Tests/TestHelperExtensions.cs
+++ b/test/Microsoft.AspNet.Diagnostics.Entity.Tests/TestHelperExtensions.cs
@@ -1,13 +1,14 @@
 // Copyright (c) Microsoft Open Technologies, Inc. All rights reserved.
 // Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
 
+using Microsoft.Data.Entity.Infrastructure;
 using Microsoft.Framework.DependencyInjection;
 
 namespace Microsoft.Data.Entity.Tests
 {
     public static class TestHelperExtensions
     {
-        public static EntityServicesBuilder AddProviderServices(this EntityServicesBuilder entityServicesBuilder)
+        public static EntityFrameworkServicesBuilder AddProviderServices(this EntityFrameworkServicesBuilder entityServicesBuilder)
         {
             return entityServicesBuilder.AddInMemoryStore();
         }