Fix tests.

Signed-off-by: Daira Hopwood <daira@jacaranda.org>