fix class type issue