Fix finding classes