I believe this may be because of dying and then being respawned while doing the puzzle, each time one key disappeared.
-
Hey there, Thank you for reporting. We'll pass this along to the developers for investigation.