Fix tests.

Whoops.