Fix for test.