Fix test case.

This is another follow-up patch to r217996.