Get the test working properly