At this stage it's okay to require a separate tx to unstake the locked tokens once they become unlocked (when the dispute is closed) - as in v1.