Fix up last failing test!