Fix breaking tests.