diff --git a/src/modules/Elsa.Agents.Persistence.EntityFrameworkCore/Feature.cs b/src/modules/Elsa.Agents.Persistence.EntityFrameworkCore/Feature.cs
index 555eb3d7b..fdfc988aa 100644
--- a/src/modules/Elsa.Agents.Persistence.EntityFrameworkCore/Feature.cs
+++ b/src/modules/Elsa.Agents.Persistence.EntityFrameworkCore/Feature.cs
@@ -1,6 +1,7 @@
 using Elsa.Agents.Persistence.Entities;
 using Elsa.Agents.Persistence.Features;
 using Elsa.EntityFrameworkCore;
+using Elsa.EntityFrameworkCore.EntityHandlers;
 using Elsa.Features.Attributes;
 using Elsa.Features.Services;
 using Microsoft.Extensions.DependencyInjection;
@@ -34,5 +35,6 @@ public class EFCoreAgentPersistenceFeature(IModule module) : PersistenceFeatureB
         AddEntityStore<ServiceDefinition, EFCoreServiceStore>();
         AddEntityStore<AgentDefinition, EFCoreAgentStore>();
         Services.AddScoped<IEntityModelCreatingHandler, SetupForOracle>();
+        Services.AddScoped<IEntityModelCreatingHandler, SetupForSqlite>();
     }
 }
\ No newline at end of file