fix class reference