Fixing a broken test.