Changed the checkpoint model