diff --git a/DigitalData.Core.Infrastructure/CRUDRepository.cs b/DigitalData.Core.Infrastructure/CRUDRepository.cs
index db8b902..dd09c1e 100644
--- a/DigitalData.Core.Infrastructure/CRUDRepository.cs
+++ b/DigitalData.Core.Infrastructure/CRUDRepository.cs
@@ -25,10 +25,10 @@ namespace DigitalData.Core.Infrastructure
         /// Initializes a new instance of the CRUDRepository with the specified DbContext.
         /// </summary>
         /// <param name="dbContext">The DbContext instance to be used by the repository.</param>
-        public CRUDRepository(TDbContext dbContext)
+        public CRUDRepository(TDbContext dbContext, DbSet<TEntity> dbSet)
         {
             _dbContext = dbContext;
-            _dbSet = dbContext.Set<TEntity>();
+            _dbSet = dbSet;
         }
 
         /// <summary>