Horrible hack to fix the LB checkpointing, needs a better fix in future.