Correct some test errors