Fix broken tests

that weren't broken a minute ago.

Signed-off-by: Jay Kuri <jayk@ionzero.com>