Fix a regression with orthogonal nudging if using checkpoints.