X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/0f2d00f9d7d5d28149707f66c719b28edfe51aed..c3d66e36e08b517f61b7f720e4a3b5a75f202cb6:/forum/models/user.py

diff --git a/forum/models/user.py b/forum/models/user.py
index 5366631..f8b8d14 100644
--- a/forum/models/user.py
+++ b/forum/models/user.py
@@ -240,12 +240,11 @@ class Activity(GenericContent):
     @property
     def node(self):
         if self.activity_type in (const.TYPE_ACTIVITY_ANSWER, const.TYPE_ACTIVITY_ASK_QUESTION,
-                const.TYPE_ACTIVITY_MARK_ANSWER, const.TYPE_ACTIVITY_UPDATE_QUESTION):
-            return self.content_object
+                const.TYPE_ACTIVITY_MARK_ANSWER, const.TYPE_ACTIVITY_COMMENT_QUESTION, const.TYPE_ACTIVITY_COMMENT_ANSWER):
+            return self.content_object.leaf
 
-        if self.activity_type in (const.TYPE_ACTIVITY_COMMENT_QUESTION,
-                const.TYPE_ACTIVITY_COMMENT_ANSWER, const.TYPE_ACTIVITY_UPDATE_ANSWER):
-            return self.content_object.node.leaf
+        if self.activity_type in (const.TYPE_ACTIVITY_UPDATE_ANSWER, const.TYPE_ACTIVITY_UPDATE_QUESTION):
+            return self.content_object.node.leaf            
             
         raise NotImplementedError()