fix heap extract tests