Yes, that's probably a better way of aligning incentives.

Plus it could be trialled on a small scale at first with just a few prisons. The main thing you'd need to watch out for is that prisons didnt just start working hard to offload their worst prisoners elsewhere.