Changed to correct class name