fixed a wrong class reference