Updated test case