Fix up tests.

This is a sweet pattern.