diff --git a/backend/alembic/versions/b058_attack_success_three_state.py b/backend/alembic/versions/b058_attack_success_three_state.py
index 732a428..c633565 100644
--- a/backend/alembic/versions/b058_attack_success_three_state.py
+++ b/backend/alembic/versions/b058_attack_success_three_state.py
@@ -25,10 +25,10 @@ def upgrade() -> None:
     op.add_column("tests", sa.Column("attack_success_new", attack_success_enum, nullable=True))
 
     op.execute(
-        "UPDATE tests SET attack_success_new = CASE "
+        "UPDATE tests SET attack_success_new = (CASE "
         "WHEN attack_success = true THEN 'successful' "
         "WHEN attack_success = false THEN 'not_successful' "
-        "ELSE NULL END"
+        f"ELSE NULL END)::{_ENUM_NAME}"
     )
 
     op.drop_column("tests", "attack_success")