Good thought, that indeed works: https://arxiv.org/abs/2310.02226