Fix breaking test case.