Fixing cross validation...