Thanks! We've changed to that from https://arxiv.org/abs/2411.04905, which is also linked there.