Ah, I see the problem that we kind of overlooked. For these puzzles you go between the normal world and the "backstage" so after solving the code puzzle you'd use the tree to go back to the normal world to progress.
Nevermind, it's an issue with the game, my bad