Update tests.

This should have been part of bd37a24.