Check that the upgrade step runs in the same transaction