modify checkpoint class.
modify checkpoint class.