Fix the test I broke

Change-Id: I3f8181f84b387ab811f5efc16c734800e3639a15