A system (20; 300) comprises: a compressor (22) having a suction port (40) and a discharge port (42); an ejector (32) having a motive flow inlet (50), a suction flow inlet (52), and an outlet (54); a separator (34) having an inlet (72), a vapor outlet (74), and a liquid outlet (76); a first heat exchanger (24); an expansion device (28); and a second heat exchanger (26; 302). Conduits and valves are positioned to provide alternative operation in: a cooling mode; a first heating mode; and a second heating mode. In the cooling mode and second heating mode, a needle (60) of the ejector is closed.