Updates to the test case