Discovered second instance of 20% bug

This patch fixes a second instance of the 20% bug that occurs when the data set gets split after label encoding.