fix class declaration