This would fail when loading a previous checkpoint (version would be 1). I think it is safe to allow version to be <= 2 and it should work for old and new checkpoints.