fixed broken test case