Fix comment display bug