Compressor control