diff --git a/PostFundManagement.Domain/Entities/Budget.cs b/PostFundManagement.Domain/Entities/Budget.cs
index 44ef49a..888463f 100644
--- a/PostFundManagement.Domain/Entities/Budget.cs
+++ b/PostFundManagement.Domain/Entities/Budget.cs
@@ -1,5 +1,6 @@
 namespace PostFundManagement.Domain.Entities;
 
+[EntityTypeConfiguration<Configuration.Budget, Budget>]
 public class Budget : Models.Budget
 {
     public virtual Project? Project { get; set; }