That was really frustrating for me too, because I did the same thing. I did find the solution they wanted later on, but the first time the cube shifted back I felt like I'd been scolded for solving the problem in a way that was perfectly viable. That clever feeling you get for solving the puzzle was let down by it, honestly.