diff --git a/src/MusicStore.Spa/Controllers/AccountController.cs b/src/MusicStore.Spa/Controllers/AccountController.cs
index 19270f19af..2b1fc5c249 100644
--- a/src/MusicStore.Spa/Controllers/AccountController.cs
+++ b/src/MusicStore.Spa/Controllers/AccountController.cs
@@ -1,7 +1,6 @@
 using System.Security.Principal;
 using System.Threading.Tasks;
 using Microsoft.AspNet.Identity;
-using Microsoft.AspNet.Identity.Security;
 using Microsoft.AspNet.Mvc;
 using Microsoft.AspNet.Mvc.ModelBinding;
 using MusicStore.Models;
diff --git a/src/MusicStore.Spa/Models/IdentityModels.cs b/src/MusicStore.Spa/Models/IdentityModels.cs
index fc1d01c570..2eb30a164f 100644
--- a/src/MusicStore.Spa/Models/IdentityModels.cs
+++ b/src/MusicStore.Spa/Models/IdentityModels.cs
@@ -1,5 +1,5 @@
 using System;
-using Microsoft.AspNet.Identity.Entity;
+using Microsoft.AspNet.Identity.EntityFramework;
 using Microsoft.Data.Entity;
 using Microsoft.Framework.OptionsModel;
 
@@ -7,7 +7,7 @@ namespace MusicStore.Models
 {
     public class ApplicationUser : User { }
 
-    public class ApplicationDbContext : IdentitySqlContext<ApplicationUser> 
+    public class ApplicationDbContext : IdentityDbContext<ApplicationUser> 
     {
         public ApplicationDbContext(IServiceProvider serviceProvider, IOptionsAccessor<IdentityDbContextOptions> optionsAccessor)
             : base(serviceProvider, optionsAccessor.Options)
diff --git a/src/MusicStore/Controllers/AccountController.cs b/src/MusicStore/Controllers/AccountController.cs
index 62f505c748..c615d52754 100644
--- a/src/MusicStore/Controllers/AccountController.cs
+++ b/src/MusicStore/Controllers/AccountController.cs
@@ -1,5 +1,4 @@
 using Microsoft.AspNet.Identity;
-using Microsoft.AspNet.Identity.Security;
 using Microsoft.AspNet.Mvc;
 using MusicStore.Models;
 using System.Security.Principal;
diff --git a/src/MusicStore/Models/IdentityModels.cs b/src/MusicStore/Models/IdentityModels.cs
index dcc779d1d8..592d0e2b1c 100644
--- a/src/MusicStore/Models/IdentityModels.cs
+++ b/src/MusicStore/Models/IdentityModels.cs
@@ -1,5 +1,5 @@
 using System;
-using Microsoft.AspNet.Identity.Entity;
+using Microsoft.AspNet.Identity.EntityFramework;
 using Microsoft.Data.Entity;
 using Microsoft.Framework.OptionsModel;
 
@@ -7,7 +7,7 @@ namespace MusicStore.Models
 {
     public class ApplicationUser : User { }
 
-    public class ApplicationDbContext : IdentitySqlContext<ApplicationUser>
+    public class ApplicationDbContext : IdentityDbContext<ApplicationUser>
     {
         public ApplicationDbContext(IServiceProvider serviceProvider, IOptionsAccessor<IdentityDbContextOptions> optionsAccessor)
                    : base(serviceProvider, optionsAccessor.Options)
diff --git a/src/MusicStore/Startup.cs b/src/MusicStore/Startup.cs
index 9f88fd4431..d92d34c1b3 100644
--- a/src/MusicStore/Startup.cs
+++ b/src/MusicStore/Startup.cs
@@ -3,7 +3,6 @@ using Microsoft.AspNet.Builder;
 using Microsoft.AspNet.Diagnostics;
 using Microsoft.AspNet.Http;
 using Microsoft.AspNet.Identity;
-using Microsoft.AspNet.Identity.Security;
 using Microsoft.AspNet.Routing;
 using Microsoft.AspNet.Security.Cookies;
 using Microsoft.Data.Entity;