remove one_hot, first try