fix class creation issue