fixed update entity clas issue