undo selected plant

Back on the Xbox 360 version of the game if you selected a plant you didn’t want you could press B to put back the recently selected plant and it was a huge quality of life feature now you have to select the plant that was selected to put it back witch works but I would also like to press B to return the last selected plant