Minor bugfix involving the improved cross validation