diff --git a/src/AspNetCore.Identity.MongoDbCore.csproj b/src/AspNetCore.Identity.MongoDbCore.csproj
index 86c1faa..5bd3c4f 100644
--- a/src/AspNetCore.Identity.MongoDbCore.csproj
+++ b/src/AspNetCore.Identity.MongoDbCore.csproj
@@ -16,7 +16,7 @@
-
+
diff --git a/src/Models/MongoIdentityRole.cs b/src/Models/MongoIdentityRole.cs
index ff96374..4d16d93 100644
--- a/src/Models/MongoIdentityRole.cs
+++ b/src/Models/MongoIdentityRole.cs
@@ -1,5 +1,4 @@
-using AspNetCore.Identity.MongoDbCore.Extensions;
-using AspNetCore.Identity.MongoDbCore.Interfaces;
+using AspNetCore.Identity.MongoDbCore.Interfaces;
using Microsoft.AspNetCore.Identity;
using MongoDbGenericRepository.Models;
using MongoDbGenericRepository.Utils;
diff --git a/src/Models/MongoIdentityUser.cs b/src/Models/MongoIdentityUser.cs
index bec8759..075a357 100644
--- a/src/Models/MongoIdentityUser.cs
+++ b/src/Models/MongoIdentityUser.cs
@@ -1,12 +1,11 @@
-using System;
-using System.Collections.Generic;
-using MongoDbGenericRepository.Models;
-using System.Linq;
-using MongoDB.Driver;
using AspNetCore.Identity.MongoDbCore.Interfaces;
using Microsoft.AspNetCore.Identity;
-using AspNetCore.Identity.MongoDbCore.Extensions;
+using MongoDB.Driver;
+using MongoDbGenericRepository.Models;
using MongoDbGenericRepository.Utils;
+using System;
+using System.Collections.Generic;
+using System.Linq;
namespace AspNetCore.Identity.MongoDbCore.Models
{
@@ -157,7 +156,7 @@ namespace AspNetCore.Identity.MongoDbCore.Models
///
public virtual MongoIdentityUser SetVersion(int version)
{
- Version = 1;
+ Version = version;
return this;
}
diff --git a/test/AspNetCore.Identity.MongoDbCore.IntegrationTests/AspNetCore.Identity.MongoDbCore.IntegrationTests.csproj b/test/AspNetCore.Identity.MongoDbCore.IntegrationTests/AspNetCore.Identity.MongoDbCore.IntegrationTests.csproj
index 6066589..5416b96 100644
--- a/test/AspNetCore.Identity.MongoDbCore.IntegrationTests/AspNetCore.Identity.MongoDbCore.IntegrationTests.csproj
+++ b/test/AspNetCore.Identity.MongoDbCore.IntegrationTests/AspNetCore.Identity.MongoDbCore.IntegrationTests.csproj
@@ -21,7 +21,7 @@
-
+