diff --git a/test/Microsoft.AspNet.Diagnostics.Entity.FunctionalTests/MigrationsEndPointMiddlewareTest.cs b/test/Microsoft.AspNet.Diagnostics.Entity.FunctionalTests/MigrationsEndPointMiddlewareTest.cs
index b752193356..863fb9ffa1 100644
--- a/test/Microsoft.AspNet.Diagnostics.Entity.FunctionalTests/MigrationsEndPointMiddlewareTest.cs
+++ b/test/Microsoft.AspNet.Diagnostics.Entity.FunctionalTests/MigrationsEndPointMiddlewareTest.cs
@@ -87,7 +87,7 @@ namespace Microsoft.AspNet.Diagnostics.Entity.Tests
                     }
                 });
 
-                using (var db = BloggingContextWithMigrations.CreateWithoutExternalServiceProvider(options))
+                using (var db = BloggingContextWithMigrations.CreateWithoutExternalServiceProvider(optionsBuilder.Options))
                 {
                     Assert.False(db.Database.AsRelational().Exists());