Fix Checkpoints.

Fix Checkpoints.