Fix failing tests.

I cocked up the tests the first time around.