If you click a Pokemon by accident in Pokeball selection, the only way to undo it would be to go through the edit mode. It'd be a nice QoL to have clicking the Pokemon again with the same ball instead remove the ball from them.