use correct class