diff --git a/test/Microsoft.AspNet.Identity.Entity.Test/TestIdentityFactory.cs b/test/Microsoft.AspNet.Identity.Entity.Test/TestIdentityFactory.cs
index 67655dab04..ba520b4cb3 100644
--- a/test/Microsoft.AspNet.Identity.Entity.Test/TestIdentityFactory.cs
+++ b/test/Microsoft.AspNet.Identity.Entity.Test/TestIdentityFactory.cs
@@ -33,10 +33,7 @@ namespace Microsoft.AspNet.Identity.Entity.Test
             var db = new IdentityContext(serviceProvider);
              
             // TODO: Recreate DB, doesn't support String ID or Identity context yet
-            if (!db.Database.Exists())
-            {
-                db.Database.Create();
-            }
+            db.Database.EnsureCreated();
 
             // TODO: CreateAsync DB?
             return db;