Fix tests.

I'm a dolt.