make the correctness tests pass