Fixed set_pxe_image_type

* When running multiple tests, the image may already have been set in
  which case the buttons failed to become active.