Cooling of compressor