Fix bug for JSON output