1 files changed, 7 insertions, 2 deletions
diff --git a/BackEnd/Timeline/Services/Mapper/MapperServiceCollectionExtensions.cs b/BackEnd/Timeline/Services/Mapper/MapperServiceCollectionExtensions.cs
index 262b2f20..0c2d0cfd 100644
--- a/BackEnd/Timeline/Services/Mapper/MapperServiceCollectionExtensions.cs
+++ b/BackEnd/Timeline/Services/Mapper/MapperServiceCollectionExtensions.cs
@@ -1,4 +1,6 @@
 using Microsoft.Extensions.DependencyInjection;
+using Timeline.Entities;
+using Timeline.Models.Http;
 
 namespace Timeline.Services.Mapper
 {
@@ -6,8 +8,11 @@ namespace Timeline.Services.Mapper
     {
         public static void AddMappers(this IServiceCollection services)
         {
-            services.AddScoped<UserMapper, UserMapper>();
-            services.AddScoped<TimelineMapper, TimelineMapper>();
+            services.AddAutoMapper(typeof(Startup).Assembly);
+            services.AddScoped<IMapper<UserEntity, HttpUser>, UserMapper>();
+            services.AddScoped<IMapper<TimelineEntity, HttpTimeline>, TimelineMapper>();
+            services.AddScoped<IMapper<TimelinePostEntity, HttpTimelinePost>, TimelineMapper>();
+            services.AddScoped<IGenericMapper, GenericMapper>();
         }
     }
 }