diff --git a/test/Microsoft.AspNet.Identity.EntityFramework.Test/SqlStoreTestBase.cs b/test/Microsoft.AspNet.Identity.EntityFramework.Test/SqlStoreTestBase.cs
index e5f3d8e52c..8cd6ddf0db 100644
--- a/test/Microsoft.AspNet.Identity.EntityFramework.Test/SqlStoreTestBase.cs
+++ b/test/Microsoft.AspNet.Identity.EntityFramework.Test/SqlStoreTestBase.cs
@@ -115,10 +115,8 @@ namespace Microsoft.AspNet.Identity.EntityFramework.Test
 
         internal static void VerifyDefaultSchema(TestDbContext dbContext)
         {
-            var sqlDb = dbContext.Database as RelationalDatabase;
-            var sqlConn = sqlDb?.Connection;
+            var sqlConn = dbContext.Database.GetDbConnection();
 
-            Assert.NotNull(sqlConn);
             using (var db = new SqlConnection(sqlConn.ConnectionString))
             {
                 db.Open();