fixed this test case