fix to checkpoint