diff --git a/snuba/migrations/autogeneration/diff.py b/snuba/migrations/autogeneration/diff.py
index f3d188f9d9..ed9147f4ad 100644
--- a/snuba/migrations/autogeneration/diff.py
+++ b/snuba/migrations/autogeneration/diff.py
@@ -78,33 +78,24 @@ def _migration_ops_to_migration(
     Given a lists of forward and backwards ops, returns a python class
     definition for the migration as a str. The migration must be non-blocking.
     """
-
-    forwards_str = (
-        "[" + ", ".join([f"operations.{repr(op)}" for op in forwards_ops]) + "]"
-    )
-    backwards_str = (
-        "[" + ", ".join([f"operations.{repr(op)}" for op in backwards_ops]) + "]"
-    )
-
     return f"""
 from typing import Sequence
 
 from snuba.clusters.storage_sets import StorageSetKey
-from snuba.migrations import operations
 from snuba.migrations.columns import MigrationModifiers
 from snuba.migrations.migration import ClickhouseNodeMigration
-from snuba.migrations.operations import OperationTarget
+from snuba.migrations.operations import AddColumn, DropColumn, OperationTarget, SqlOperation
 from snuba.utils import schemas
 from snuba.utils.schemas import Column
 
 class Migration(ClickhouseNodeMigration):
     blocking = False
 
-    def forwards_ops(self) -> Sequence[operations.SqlOperation]:
-        return {forwards_str}
+    def forwards_ops(self) -> Sequence[SqlOperation]:
+        return {repr(forwards_ops)}
 
-    def backwards_ops(self) -> Sequence[operations.SqlOperation]:
-        return {backwards_str}
+    def backwards_ops(self) -> Sequence[SqlOperation]:
+        return {repr(backwards_ops)}
 """
 
 
diff --git a/tests/migrations/autogeneration/test_generate_python_migration.py b/tests/migrations/autogeneration/test_generate_python_migration.py
index f0336eaeca..9ab7e4987b 100644
--- a/tests/migrations/autogeneration/test_generate_python_migration.py
+++ b/tests/migrations/autogeneration/test_generate_python_migration.py
@@ -51,10 +51,9 @@ def test_add_column() -> None:
 from typing import Sequence
 
 from snuba.clusters.storage_sets import StorageSetKey
-from snuba.migrations import operations
 from snuba.migrations.columns import MigrationModifiers
 from snuba.migrations.migration import ClickhouseNodeMigration
-from snuba.migrations.operations import OperationTarget
+from snuba.migrations.operations import AddColumn, DropColumn, OperationTarget, SqlOperation
 from snuba.utils import schemas
 from snuba.utils.schemas import Column
 
@@ -62,23 +61,23 @@ def test_add_column() -> None:
 class Migration(ClickhouseNodeMigration):
     blocking = False
 
-    def forwards_ops(self) -> Sequence[operations.SqlOperation]:
+    def forwards_ops(self) -> Sequence[SqlOperation]:
         return [
-            operations.AddColumn(
+            AddColumn(
                 storage_set=StorageSetKey.EVENTS,
                 table_name="errors_local",
                 column=Column("newcol1", schemas.DateTime(modifiers=None)),
                 after="timestamp",
                 target=OperationTarget.LOCAL,
             ),
-            operations.AddColumn(
+            AddColumn(
                 storage_set=StorageSetKey.EVENTS,
                 table_name="errors_dist",
                 column=Column("newcol1", schemas.DateTime(modifiers=None)),
                 after="timestamp",
                 target=OperationTarget.DISTRIBUTED,
             ),
-            operations.AddColumn(
+            AddColumn(
                 storage_set=StorageSetKey.EVENTS,
                 table_name="errors_local",
                 column=Column(
@@ -98,7 +97,7 @@ def forwards_ops(self) -> Sequence[operations.SqlOperation]:
                 after="event_id",
                 target=OperationTarget.LOCAL,
             ),
-            operations.AddColumn(
+            AddColumn(
                 storage_set=StorageSetKey.EVENTS,
                 table_name="errors_dist",
                 column=Column(
@@ -120,27 +119,27 @@ def forwards_ops(self) -> Sequence[operations.SqlOperation]:
             ),
         ]
 
-    def backwards_ops(self) -> Sequence[operations.SqlOperation]:
+    def backwards_ops(self) -> Sequence[SqlOperation]:
         return [
-            operations.DropColumn(
+            DropColumn(
                 storage_set=StorageSetKey.EVENTS,
                 table_name="errors_dist",
                 column_name="newcol2",
                 target=OperationTarget.DISTRIBUTED,
             ),
-            operations.DropColumn(
+            DropColumn(
                 storage_set=StorageSetKey.EVENTS,
                 table_name="errors_local",
                 column_name="newcol2",
                 target=OperationTarget.LOCAL,
             ),
-            operations.DropColumn(
+            DropColumn(
                 storage_set=StorageSetKey.EVENTS,
                 table_name="errors_dist",
                 column_name="newcol1",
                 target=OperationTarget.DISTRIBUTED,
             ),
-            operations.DropColumn(
+            DropColumn(
                 storage_set=StorageSetKey.EVENTS,
                 table_name="errors_local",
                 column_name="newcol1",