I also found myself thinking that if I want to hold the magnet as I enter a level, I want to hold it as I exit a level, also, and how messy that could potentially be for completing a level given all the different positions the magnet needs to be in as you collect keys across levels; but then I remembered the way Celeste handled some strawberries; so maybe the robot can collect the key whenever/however the player can accomplish it, but then the level ends when you reconnect with the magnet; I'm not sure! Just a few stray thoughts!