The tutorial section made me think that the level structure would involve collecting a key and then getting your magnet buddy back to the door but in the levels you just get the key and the level ends. I almost feel like it would tie the theme together if you had to exit the level with your magnet, otherwise it feels like letting Yoshi fall in a pit while you jump to the end of the level.