Register the right class