Fix unit tests

Forgot to update them when doing ag/9847483

Test: mma
Change-Id: I0c1c769c001adc1b3520e985a120cce8b5ed8f4c
1 file changed