Fixed model relation logic on comment.