fix checkpoint checking + update test