X-Git-Url: https://git.openstreetmap.org/osqa.git/blobdiff_plain/77c34a5005c5e159e7795c985888162b88609d5d..0f2d00f9d7d5d28149707f66c719b28edfe51aed:/forum/models/meta.py

diff --git a/forum/models/meta.py b/forum/models/meta.py
index 538d6a2..7e82137 100644
--- a/forum/models/meta.py
+++ b/forum/models/meta.py
@@ -22,7 +22,7 @@ class Vote(MetaContent, UserContent):
         return '[%s] voted at %s: %s' %(self.user, self.voted_at, self.vote)
 
     def _update_post_vote_count(self, diff):
-        post = self.content_object
+        post = self.node
         field = self.vote == 1 and 'vote_up_count' or 'vote_down_count'
         post.__dict__[field] = post.__dict__[field] + diff
         post.save()
@@ -62,7 +62,7 @@ class FlaggedItem(MetaContent, UserContent):
         return '[%s] flagged at %s' %(self.user, self.flagged_at)
 
     def _update_post_flag_count(self, diff):
-        post = self.content_object
+        post = self.node
         post.offensive_flag_count = post.offensive_flag_count + diff
         post.save()
 
@@ -89,7 +89,7 @@ class Comment(MetaContent, UserContent, DeletableContent):
         db_table = u'comment'
 
     def _update_post_comment_count(self, diff):
-        post = self.content_object
+        post = self.node.leaf
         post.comment_count = post.comment_count + diff
         post.save()