def upgrade(): with op.batch_alter_table('pages') as batch_op: batch_op.drop_column('is_html') batch_op.drop_column('use_layout_header')