修改获取class的方式