fix a UT function
Merge pull request #1605 from sijchen/fix_ut1

fix a UT function