fix class type hinting