Yep, I found this issue while watching a streamer play the game.
The puzzle should play correctly on the first run, but if you restart the level with something pressing the button it'll break. Work around for now is to exit to the level select and reload it.