In the Shadowfell cutscene, Balthazar is flying. So it makes sense that he can simply fly back onto the prisoner's rock. (Same situation in Cazador fight.)

Sounds like patched to be bit more consistent.