Fix regressed test case
Merge pull request #3202 from ke-yu/fix_testcase

Fix regressed test case