Fix tests.

Signed-off-by: Austin Seipp <aseipp@pobox.com>