Use the SampleSet key or the first Sample key in checkpoint wrapper Closes #190 See merge request !319