fix a class name[bug]

fix a class name[bug]