Fixing python test failure.

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