Fix bug

- fix bug in the cross validation func