diff --git a/src/modules/Elsa.MongoDb/Modules/Management/CreateIndices.cs b/src/modules/Elsa.MongoDb/Modules/Management/CreateIndices.cs
index d371d4ad4..4344247b5 100644
--- a/src/modules/Elsa.MongoDb/Modules/Management/CreateIndices.cs
+++ b/src/modules/Elsa.MongoDb/Modules/Management/CreateIndices.cs
@@ -27,7 +27,7 @@ internal class CreateIndices : IHostedService
 
     private static Task CreateWorkflowDefinitionIndices(IServiceScope serviceScope, CancellationToken cancellationToken)
     {
-        var workflowDefinitionCollection = serviceScope.ServiceProvider.GetService<MongoCollectionBase<WorkflowDefinition>>();
+        var workflowDefinitionCollection = serviceScope.ServiceProvider.GetService<IMongoCollection<WorkflowDefinition>>();
         if (workflowDefinitionCollection == null) return Task.CompletedTask;
 
         return IndexHelpers.CreateAsync(
@@ -51,7 +51,7 @@ internal class CreateIndices : IHostedService
 
     private static Task CreateWorkflowInstanceIndices(IServiceScope serviceScope, CancellationToken cancellationToken)
     {
-        var workflowInstanceCollection = serviceScope.ServiceProvider.GetService<MongoCollectionBase<WorkflowInstance>>();
+        var workflowInstanceCollection = serviceScope.ServiceProvider.GetService<IMongoCollection<WorkflowInstance>>();
         if (workflowInstanceCollection == null) return Task.CompletedTask;
 
         return IndexHelpers.CreateAsync(
@@ -86,4 +86,5 @@ internal class CreateIndices : IHostedService
                     },
                     cancellationToken));
     }
-}
\ No newline at end of file
+}
+