error in visualize_json_file.py

Hi, thank you for your work~
 
in line 23:
classes = obj["classes"]

but in coco defination, there is no "classes" attr.
{
"info": info, "images": [image], "annotations": [annotation], "licenses": [license],
}
do you mean "categories" here?