Fix checkpoints unit test.