get_class fix