changing signature of in-memory checkpoint restart function