Fixing broken test.

Reviewed at https://reviews.apache.org/r/25673/