Fix get_class