fixed wrong class