diff --git a/tweak/Reddit.xm b/tweak/Reddit.xm
index 10f1dbf..2c58e6f 100755
--- a/tweak/Reddit.xm
+++ b/tweak/Reddit.xm
@@ -185,11 +185,14 @@ int secondVersionPart = 0;
 	[comment setBodyText:body];
 	[comment setBodyRichTextAttributed:bodyMutableAttributedText];
 
-	if (secondVersionPart <= 39) {
+	if ((firstVersionPart == 2020 && secondVersionPart <= 39) || firstVersionPart == 4) {
 		[comment setBodyAttributedText:bodyMutableAttributedText];
 	}
 
-	[[commentTreeNode commentTreeHeaderNode] updateContentViewsForData:comment];
+	CommentTreeHeaderNode *headerNode = [commentTreeNode commentTreeHeaderNode];
+
+	[headerNode updateContentViewsForData:comment];
+	[headerNode configureNodes];
 
 	[bodyMutableAttributedText release];
 }