FIx bug on comment display