removing the deprecated class reference