Use correct class