diff --git a/TIAM.Database/DbContexts/Users/UserDbContext.cs b/TIAM.Database/DbContexts/Users/UserDbContext.cs
index e00d5536..4a5f0f55 100644
--- a/TIAM.Database/DbContexts/Users/UserDbContext.cs
+++ b/TIAM.Database/DbContexts/Users/UserDbContext.cs
@@ -24,6 +24,8 @@ namespace TIAM.Database.DbContexts.Users
         public DbSet<Company> Companies { get; set; }
 
         public DbSet<Transfer> Transfers { get; set; }
+        public DbSet<TransferToDriver> TransferToDrivers { get; set; }
+
         public DbSet<TransferDestinationToProduct> TransferDestinationToProducts { get; set; }
 
         public DbSet<TransferDestination> TransferDestinations { get; set; }
@@ -86,6 +88,5 @@ namespace TIAM.Database.DbContexts.Users
             //        //builderUserProductJsonDetail.OwnsMany(userProductJsonDetail => userProductJsonDetail.Cars2);
             //    });
         }
-        
     }
 }
diff --git a/TIAM.Database/DbSets/Transfers/TransferToDriverDbSetExtensions.cs b/TIAM.Database/DbSets/Transfers/TransferToDriverDbSetExtensions.cs
index a73905d4..d89e6acc 100644
--- a/TIAM.Database/DbSets/Transfers/TransferToDriverDbSetExtensions.cs
+++ b/TIAM.Database/DbSets/Transfers/TransferToDriverDbSetExtensions.cs
@@ -16,7 +16,7 @@ public static class TransferToDriverDbSetExtensions
     public static IQueryable<TransferToDriver> GetTransferToDriversByTransferId(this ITransferToDriverDbSet ctx, Guid transferId, bool autoInclude = true)
         => ctx.TransferToDrivers.Where(x => x.TransferId == transferId);
 
-    public static bool AddTransferToDriver(this IDriverDbSet ctx, TransferToDriver transferToDriver)
+    public static bool AddTransferToDriver(this ITransferDbSet ctx, TransferToDriver transferToDriver)
     {
         var transfer = ctx.GetTransferById(transferToDriver.TransferId)!;
         transfer.TransferStatusType = TransferStatusType.AssignedToDriver;
@@ -24,13 +24,13 @@ public static class TransferToDriverDbSetExtensions
         return ctx.TransferToDrivers.Add(transferToDriver).State == EntityState.Added;
     }
     
-    private static bool RemoveTransferToDriver(this ITransferToDriverDbSet ctx, TransferToDriver transferToDriver)
+    private static bool RemoveTransferToDriver(this ITransferDbSet ctx, TransferToDriver transferToDriver)
     {
         //TODO: TransferStatusType change, ha nincs sofőr a törlés után! - J.
         return ctx.TransferToDrivers.Remove(transferToDriver).State == EntityState.Deleted;
     }
 
-    public static bool RemoveTransferToDriver(this ITransferToDriverDbSet ctx, Guid transferToDriverId)
+    public static bool RemoveTransferToDriver(this ITransferDbSet ctx, Guid transferToDriverId)
     {
         var transferToDriver = ctx.GetTransferToDriverById(transferToDriverId);
         return transferToDriver == null || ctx.RemoveTransferToDriver(transferToDriver);