Fixing build break.

Couple of tests changes escaped in https://reviews.apache.org/r/29165

Testing Done:
./pants src/test/python:all

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