Cooling of pump drives