Fix missed classes use