Add local_ax_count to the list of parameters taken by calculate_element_region.