fix checkpoint