Correct some tests