diff --git a/wax-prosemirror-services/src/CommentsService/components/ui/comments/CommentBox.js b/wax-prosemirror-services/src/CommentsService/components/ui/comments/CommentBox.js
index 3aad3597bb7351218f343ee2c6130beca397f99a..5519a90050bc3fbed599848c257070b7e4983a03 100644
--- a/wax-prosemirror-services/src/CommentsService/components/ui/comments/CommentBox.js
+++ b/wax-prosemirror-services/src/CommentsService/components/ui/comments/CommentBox.js
@@ -42,7 +42,7 @@ const Head = styled.div`
   justify-content: flex-end;
   padding: 8px 16px 0;
 
-  ${override('Wax.CommentHead')}
+  ${override('Wax.CommentResolveWrapper')}
 `;
 
 const Resolve = styled.button`
diff --git a/wax-prosemirror-services/src/CommentsService/components/ui/comments/CommentItem.js b/wax-prosemirror-services/src/CommentsService/components/ui/comments/CommentItem.js
index 6e273c5cec07887ed815664c6640fde9439104e7..01d99b350a1b987caa8ca4aaf346d59cb6c6f48f 100644
--- a/wax-prosemirror-services/src/CommentsService/components/ui/comments/CommentItem.js
+++ b/wax-prosemirror-services/src/CommentsService/components/ui/comments/CommentItem.js
@@ -13,7 +13,7 @@ const Head = styled.div`
   font-size: 14px;
   margin-bottom: ${grid(2)};
 
-  ${override('Wax.CommentHead')}
+  ${override('Wax.CommentInfoWrapper')}
 `;
 
 const Name = styled.div`