Fix invalid class