Return the correct class