Minor change

Corrected update checkpoint