I'm having trouble with that too. I wouldn't mind as much if it went back to the puzzle after, just a mild annoyance at that point for losing a few seconds...