Updates to the test