fix wrong class