diff --git a/DrinkRateAPI/Contexts/ApplicationDbContext.cs b/DrinkRateAPI/Contexts/ApplicationDbContext.cs
index 0c50f6a..604ff93 100644
--- a/DrinkRateAPI/Contexts/ApplicationDbContext.cs
+++ b/DrinkRateAPI/Contexts/ApplicationDbContext.cs
@@ -33,6 +33,8 @@ public class ApplicationDbContext : IdentityDbContext<DbApplicationUser>
 
     protected override void OnModelCreating(ModelBuilder modelBuilder)
     {
+        base.OnModelCreating(modelBuilder);
+        
         // Company
         modelBuilder.Entity<DbCompany>(entity =>
         {
@@ -216,6 +218,5 @@ public class ApplicationDbContext : IdentityDbContext<DbApplicationUser>
                 .WithMany();
         });
         
-        base.OnModelCreating(modelBuilder);
     }
 }
\ No newline at end of file
diff --git a/DrinkRateAPI/Migrations/20250809141112_init.Designer.cs b/DrinkRateAPI/Migrations/20250809145003_init.Designer.cs
similarity index 99%
rename from DrinkRateAPI/Migrations/20250809141112_init.Designer.cs
rename to DrinkRateAPI/Migrations/20250809145003_init.Designer.cs
index 370ebe9..a4e6100 100644
--- a/DrinkRateAPI/Migrations/20250809141112_init.Designer.cs
+++ b/DrinkRateAPI/Migrations/20250809145003_init.Designer.cs
@@ -12,7 +12,7 @@ using Npgsql.EntityFrameworkCore.PostgreSQL.Metadata;
 namespace DrinkRateAPI.Migrations
 {
     [DbContext(typeof(ApplicationDbContext))]
-    [Migration("20250809141112_init")]
+    [Migration("20250809145003_init")]
     partial class init
     {
         /// <inheritdoc />
diff --git a/DrinkRateAPI/Migrations/20250809141112_init.cs b/DrinkRateAPI/Migrations/20250809145003_init.cs
similarity index 100%
rename from DrinkRateAPI/Migrations/20250809141112_init.cs
rename to DrinkRateAPI/Migrations/20250809145003_init.cs
diff --git a/DrinkRateAPI/Program.cs b/DrinkRateAPI/Program.cs
index 6634693..6078b3c 100644
--- a/DrinkRateAPI/Program.cs
+++ b/DrinkRateAPI/Program.cs
@@ -1,4 +1,5 @@
 using DrinkRateAPI.Contexts;
+using DrinkRateAPI.DbEntities;
 using Microsoft.AspNetCore.Identity;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.OpenApi.Models;
@@ -11,7 +12,7 @@ builder.Services.AddControllers();
 // Learn more about configuring Swagger/OpenAPI at https://aka.ms/aspnetcore/swashbuckle
 builder.Services.AddEndpointsApiExplorer();
 builder.Services.AddAuthorization();
-builder.Services.AddIdentityApiEndpoints<IdentityUser>()
+builder.Services.AddIdentityApiEndpoints<DbApplicationUser>()
     .AddEntityFrameworkStores<ApplicationDbContext>();
 
 builder.Services.AddSwaggerGen(c =>
@@ -60,7 +61,7 @@ if (app.Environment.IsDevelopment())
     app.MapSwagger().RequireAuthorization();
 }
 
-app.MapIdentityApi<IdentityUser>();
+app.MapIdentityApi<DbApplicationUser>();
 
 app.UseHttpsRedirection();