Update cross validation to take advantage of new fit signature.