I think adding this should be optional, based on an argument (and this query should be added inside ``$queries`` returned from ``$this->execute($direction, true);`` based on this extra argument (forwarding it), to handle the case of batch migrations as well