fix class edit