fixed reference to wrong class