fix bug of assign data
Merge pull request #606 from huili2/PredUT_fix

fix bug of assign data of UT code