diff --git a/app/models/budget/ballot/line.rb b/app/models/budget/ballot/line.rb
index 83dd240b8..85434ad2d 100644
--- a/app/models/budget/ballot/line.rb
+++ b/app/models/budget/ballot/line.rb
@@ -34,9 +34,9 @@ class Budget
       private
 
         def set_denormalized_ids
-          self.heading_id ||= self.investment.heading_id
-          self.group_id   ||= self.investment.group_id
-          self.budget_id  ||= self.investment.budget_id
+          self.heading_id ||= self.investment.try(:heading_id)
+          self.group_id   ||= self.investment.try(:group_id)
+          self.budget_id  ||= self.investment.try(:budget_id)
         end
     end
   end
diff --git a/app/models/budget/investment.rb b/app/models/budget/investment.rb
index d16e8ef36..23586ec3a 100644
--- a/app/models/budget/investment.rb
+++ b/app/models/budget/investment.rb
@@ -215,8 +215,8 @@ class Budget
     private
 
       def set_denormalized_ids
-        self.group_id ||= self.heading.group_id
-        self.budget_id ||= self.heading.group.budget_id
+        self.group_id ||= self.heading.try(:group_id)
+        self.budget_id ||= self.heading.try(:group).try(:budget_id)
       end
   end
 end