Fix weirdness when completing a Checkpoint where UI wasn't updating