Fix comments UI