Fix broken tests.

Forgot to run tests at last commit.