Fix for unit test.