Fixed the include_prob comment.