There was no stunt double, it was all Baltz in the scene, and he trained hard to master the routine. It started with a call from a producer to start doing Pilates, because he would need core strength.