diff --git a/src/ReC.API/Controllers/RecActionController.cs b/src/ReC.API/Controllers/RecActionController.cs
index 30e4c38..5d49c08 100644
--- a/src/ReC.API/Controllers/RecActionController.cs
+++ b/src/ReC.API/Controllers/RecActionController.cs
@@ -3,7 +3,6 @@ using Microsoft.AspNetCore.Mvc;
 using ReC.API.Models;
 using ReC.Application.RecActions.Commands;
 using ReC.Application.RecActions.Queries;
-using System.IO;
 using System.Text.Json;
 
 namespace ReC.API.Controllers;
diff --git a/src/ReC.API/Program.cs b/src/ReC.API/Program.cs
index 7b679e8..0d0204a 100644
--- a/src/ReC.API/Program.cs
+++ b/src/ReC.API/Program.cs
@@ -16,7 +16,7 @@ builder.Services.AddRecServices(options =>
 
 builder.Services.AddRecInfrastructure(options =>
 {
-    options.ConfigureDbContext((dbContextOpt) =>
+    options.ConfigureDbContext((provider, dbContextOpt) =>
     {
         var connectionString = builder.Configuration.GetConnectionString("Default") 
                                ?? throw new InvalidOperationException("Connection string is not found.");
diff --git a/src/ReC.Infrastructure/DependencyInjection.cs b/src/ReC.Infrastructure/DependencyInjection.cs
index be0e7c6..309c59b 100644
--- a/src/ReC.Infrastructure/DependencyInjection.cs
+++ b/src/ReC.Infrastructure/DependencyInjection.cs
@@ -31,9 +31,9 @@ public static class DependencyInjection
 
     public class ConfigurationOptions
     {
-        internal Action<DbContextOptionsBuilder>? DbContextOptionsAction { get; private set; }
+        internal Action<IServiceProvider, DbContextOptionsBuilder>? DbContextOptionsAction { get; private set; }
 
-        public ConfigurationOptions ConfigureDbContext(Action<DbContextOptionsBuilder> optionsAction)
+        public ConfigurationOptions ConfigureDbContext(Action<IServiceProvider, DbContextOptionsBuilder> optionsAction)
         {
             DbContextOptionsAction = optionsAction;
             return this;