Fix tests.

(Need coffee.)