Import correct class.