Fix incorrect checkpoint

Thx richard