diff --git a/app/models/budget.rb b/app/models/budget.rb
index d19b38959..abffbf799 100644
--- a/app/models/budget.rb
+++ b/app/models/budget.rb
@@ -20,5 +20,4 @@ class Budget < ActiveRecord::Base
     phase == "finished"
   end
 
-end
-
+end
\ No newline at end of file
diff --git a/app/models/budget/heading.rb b/app/models/budget/heading.rb
index d6c7aa127..e79c5b7c8 100644
--- a/app/models/budget/heading.rb
+++ b/app/models/budget/heading.rb
@@ -3,6 +3,8 @@ class Budget
     belongs_to :budget
     belongs_to :geozone
 
+    has_many :investments
+
     validates :budget_id, presence: true
     validates :name, presence: true
     validates :price, presence: true
diff --git a/app/models/budget/investment.rb b/app/models/budget/investment.rb
index 4fc7f4485..e1e193a81 100644
--- a/app/models/budget/investment.rb
+++ b/app/models/budget/investment.rb
@@ -17,7 +17,6 @@ class Budget
     has_many :valuation_assignments, dependent: :destroy
     has_many :valuators, through: :valuation_assignments
     has_many :comments, as: :commentable
-    has_many :headings, dependent: :destroy
 
     validates :title, presence: true
     validates :author, presence: true