Fix play and store trajectories