Fix tests, oops.