More EZH fixes