diff --git a/src/presentation/DigitalData.MessagingService.API/Program.cs b/src/presentation/DigitalData.MessagingService.API/Program.cs
index a5b4fb2..779442d 100644
--- a/src/presentation/DigitalData.MessagingService.API/Program.cs
+++ b/src/presentation/DigitalData.MessagingService.API/Program.cs
@@ -1,5 +1,5 @@
 using DigitalData.MessagingService.API.Middleware;
-using DigitalData.MessagingService.Application;
+using DigitalData.MessagingService.Application.DependencyInjection;
 using DigitalData.MessagingService.Infrastructure;
 using DigitalData.MessagingService.Infrastructure.Services.Background;
 using HealthChecks.UI.Client;
@@ -68,7 +68,7 @@ try
     builder.Services.AddMessagingServiceApplication(builder.Configuration);
 
     // Register Infrastructure layer (RabbitMQ, Repositories, etc.)
-    builder.Services.AddInfrastructure(builder.Configuration);
+    builder.Services.AddInfrastructure(cfg => cfg.AddRabbitMqConnectionFactory(builder.Configuration));
 
     // Health checks — each check corresponds to a singleton registered in the Infrastructure layer
     builder.Services.AddHealthChecks()