Fix tests.

This is why we test!