Use correct class.