fixed broken test.