minor fix

added headers for one hot