I think this change wants a self.needs_halo_update = True as well