Modify test code a little.

Internal issue number: 2195990
3 files changed