diff --git a/ereuse_devicehub/migrations/versions/b4bd1538bad5_update_live.py b/ereuse_devicehub/migrations/versions/b4bd1538bad5_update_live.py
index ffaa57ea..f340df6a 100644
--- a/ereuse_devicehub/migrations/versions/b4bd1538bad5_update_live.py
+++ b/ereuse_devicehub/migrations/versions/b4bd1538bad5_update_live.py
@@ -27,7 +27,7 @@ def get_inv():
 def upgrade():
     # op.execute("COMMIT")
     op.execute("ALTER TYPE snapshotsoftware ADD VALUE 'WorkbenchDesktop'")
-    SOFTWARE = sa.Enum(
+    SOFTWARE = sa.dialects.postgresql.ENUM(
         'Workbench',
         'WorkbenchAndroid',
         'AndroidApp',