diff --git a/CTFd/plugins/migrations.py b/CTFd/plugins/migrations.py
index 71b3995a..cf4ddea7 100644
--- a/CTFd/plugins/migrations.py
+++ b/CTFd/plugins/migrations.py
@@ -22,12 +22,14 @@ def get_all_tables(op):
     return tables
 
 
-def get_columns_for_table(op, table_name):
+def get_columns_for_table(op, table_name, names_only=False):
     """
     Function to list the columns in a table from a migration
     """
     inspector = SQLAInspect(op.get_bind())
     columns = inspector.get_columns(table_name)
+    if names_only is True:
+        columns = [c["name"] for c in columns]
     return columns