Minor change to test case