Fixing broken test.