fixed issue splitting train/test files