fix bug: when training set includes only one table