updated the test and the return from the correct() method