Provisional fix for unit test.