Forgot one thing in the checkpoint fix...