Thanks @millad456!
Oh that makes a lot more sense.
It's interesting, I solved the last puzzle by throwing the magnet into the hole to stop the block from pressing the switch. This required a very precise aim and a couple of tricky jumps, and also didn't feel like the intended solution.
