fixed wrong class reference