fix class ..