diff --git a/migrate/tests/changeset/test_changeset.py b/migrate/tests/changeset/test_changeset.py index 2b50922..d1b34cd 100644 --- a/migrate/tests/changeset/test_changeset.py +++ b/migrate/tests/changeset/test_changeset.py @@ -149,7 +149,10 @@ class TestAddDropColumn(fixture.DB): def add_func(col): return self.table.create_column(col) def drop_func(col): - self.table.c.remove(col) + if SQLA_07: + self.table._columns.remove(col) + else: + self.table.c.remove(col) return self.table.drop_column(col.name) self.run_(add_func, drop_func)