Fix up remaining test.