Fixes second instance of 20% bug when spliting optimization and test data sets.