Import the right class