diff --git a/src/HeartTrackAPI/Utils/AppBootstrap.cs b/src/HeartTrackAPI/Utils/AppBootstrap.cs
index 6ad72db..e15d89f 100644
--- a/src/HeartTrackAPI/Utils/AppBootstrap.cs
+++ b/src/HeartTrackAPI/Utils/AppBootstrap.cs
@@ -105,16 +105,23 @@ public class AppBootstrap(IConfiguration configuration)
         services.AddIdentityApiEndpoints<AthleteEntity>()
             .AddEntityFrameworkStores<TrainingStubbedContext>();*/
 
-        services.AddIdentity<AthleteEntity, IdentityRole<int>>(options =>
-            {
-                options.Password.RequireDigit = true;
-                options.Password.RequireLowercase = true;
-                options.Password.RequireUppercase = true;
-                options.Password.RequireNonAlphanumeric = true;
-                options.Password.RequiredLength = 8;
-            })
-            .AddEntityFrameworkStores<TrainingStubbedContext>();
+        var identityBuilder = services.AddIdentity<AthleteEntity, IdentityRole<int>>(options =>
+        {
+            options.Password.RequireDigit = true;
+            options.Password.RequireLowercase = true;
+            options.Password.RequireUppercase = true;
+            options.Password.RequireNonAlphanumeric = true;
+            options.Password.RequiredLength = 8;
+        });
 
+        if (Environment.GetEnvironmentVariable("TYPE") == "BDD")
+        {
+            identityBuilder.AddEntityFrameworkStores<HeartTrackContext>();
+        }
+        else
+        {
+            identityBuilder.AddEntityFrameworkStores<TrainingStubbedContext>();
+        }
 
         services.AddAuthentication(options =>
         {