diff --git a/migrations/versions/f5c595e19c07_.py b/migrations/versions/f5c595e19c07_.py
index c7b10ff..9962160 100644
--- a/migrations/versions/f5c595e19c07_.py
+++ b/migrations/versions/f5c595e19c07_.py
@@ -18,15 +18,13 @@
 
 def upgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    op.add_column('publications', sa.Column('nif_assoc', sa.Integer(), nullable=True))
     op.drop_column('publications', 'nif_funded')
-    op.drop_column('publications', 'nif_likelihood')
+    op.alter_column('publications', 'nif_likelihood', new_column_name='nif_assoc')
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    op.add_column('publications', sa.Column('nif_likelihood', sa.INTEGER(), nullable=True))
     op.add_column('publications', sa.Column('nif_funded', sa.BOOLEAN(), nullable=True))
-    op.drop_column('publications', 'nif_assoc')
+    op.alter_column('publications', 'nif_assoc', new_column_name='nif_likelihood')
     # ### end Alembic commands ###