Either way would work. As long as the socket is the conventional type with the screws that tighten down on the wires as opposed to something that is not seviceable. I would take it apart and shorten the cord from the socket end, that way you don't have to buy a new plug. Make sure you pay attention to the polarity of the cord. Hot goes to the center of the socket and neutral to the outside threads of the bulb.