diff --git a/WorkFlow.Infrastructure/WFDBContext.cs b/WorkFlow.Infrastructure/WFDBContext.cs
index 5376fca..a9288cf 100644
--- a/WorkFlow.Infrastructure/WFDBContext.cs
+++ b/WorkFlow.Infrastructure/WFDBContext.cs
@@ -5,5 +5,14 @@ namespace WorkFlow.Infrastructure
 {
     public class WFDBContext(DbContextOptions options) : DbContext(options)
     {
+        public DbSet<Config> Configs { get; set; }
+
+        public DbSet<ProfControlsTf> ProfControlsTf { get; set; }
+
+        public DbSet<Profile> Profiles { get; set; }
+
+        public DbSet<ProfileObjState> ProfileObjStates { get; set; }
+
+        public DbSet<WfState> WfStates { get; set; }
     }
 }
\ No newline at end of file