2 files changed, 3 insertions, 2 deletions
diff --git a/Timeline/Authenticate/Attribute.cs b/Timeline/Authenticate/Attribute.cs
index 1875a21b..239a2a1c 100644
--- a/Timeline/Authenticate/Attribute.cs
+++ b/Timeline/Authenticate/Attribute.cs
@@ -1,5 +1,5 @@
 using Microsoft.AspNetCore.Authorization;
-using Timeline.Models;
+using Timeline.Entities;
 
 namespace Timeline.Authenticate
 {
diff --git a/Timeline/Authenticate/AuthHandler.cs b/Timeline/Authenticate/AuthHandler.cs
index 5b26716d..f9409c1a 100644
--- a/Timeline/Authenticate/AuthHandler.cs
+++ b/Timeline/Authenticate/AuthHandler.cs
@@ -7,6 +7,7 @@ using System.Linq;
 using System.Security.Claims;
 using System.Text.Encodings.Web;
 using System.Threading.Tasks;
+using Timeline.Models;
 using Timeline.Services;
 
 namespace Timeline.Authenticate
@@ -80,7 +81,7 @@ namespace Timeline.Authenticate
 
                 var identity = new ClaimsIdentity(AuthConstants.Scheme);
                 identity.AddClaim(new Claim(identity.NameClaimType, userInfo.Username, ClaimValueTypes.String));
-                identity.AddClaims(Entities.UserUtility.IsAdminToRoleArray(userInfo.Administrator).Select(role => new Claim(identity.RoleClaimType, role, ClaimValueTypes.String)));
+                identity.AddClaims(UserUtility.IsAdminToRoleArray(userInfo.Administrator).Select(role => new Claim(identity.RoleClaimType, role, ClaimValueTypes.String)));
 
                 var principal = new ClaimsPrincipal();
                 principal.AddIdentity(identity);