Fixed broken test.