fix comments UI issue