Update the test case a bit