Fix broke tests.