Fix bug for comments.

Signed-off-by: Jiahe Jou <joujiahe@gmail.com>