Yeah, I think the puzzle is stronger if you have to engage with this mechanic. And since the difficulty level/state space is low anyway, changes like this aren't too dangerous.
