Fix breaking test.