fix custom folds regression from last commit.