fix broken test.