fix wrong class definition