Fixes unit test.