fix add checkpoint