remove one_hot, first try

I think the test should be updated.