fix get_class error