Fix broken TB4 test.

Change-Id: I0e1c319738dea239026ca7a366d5649823183bc0