diff --git a/arkindex/dataimport/migrations/0007_worker_workerversion.py b/arkindex/dataimport/migrations/0007_worker_workerversion.py
index 0f7d0fc81d823be5224e1b3fe4ce4ff29a59336a..8190e417d9d392577da42bfd916d5da4e11f64b9 100644
--- a/arkindex/dataimport/migrations/0007_worker_workerversion.py
+++ b/arkindex/dataimport/migrations/0007_worker_workerversion.py
@@ -1,14 +1,13 @@
 # Generated by Django 2.2.11 on 2020-05-19 14:55
 
 import uuid
+from enum import Enum
 
 import django.contrib.postgres.fields.jsonb
 import django.db.models.deletion
 import enumfields.fields
 from django.db import migrations, models
 
-import arkindex_common.ml_tool
-
 
 class Migration(migrations.Migration):
 
@@ -23,7 +22,7 @@ class Migration(migrations.Migration):
                 ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
                 ('name', models.CharField(max_length=100)),
                 ('slug', models.CharField(max_length=100)),
-                ('type', enumfields.fields.EnumField(enum=arkindex_common.ml_tool.MLToolType, max_length=50)),
+                ('type', enumfields.fields.EnumField(enum=Enum('MLToolType', ''), max_length=50)),
                 ('repository', models.ForeignKey(
                     on_delete=django.db.models.deletion.CASCADE,
                     related_name='workers',
diff --git a/arkindex/documents/migrations/0001_initial.py b/arkindex/documents/migrations/0001_initial.py
index 81c04e49b9345f413405b4ee6adb61d7176f2782..3e3e3aaac52cd4010bb0e80dc771463a3f8543bd 100644
--- a/arkindex/documents/migrations/0001_initial.py
+++ b/arkindex/documents/migrations/0001_initial.py
@@ -6,12 +6,12 @@ import django.contrib.postgres.fields.hstore
 import django.db.models.deletion
 import enumfields.fields
 from django.db import migrations, models
+from enumfields import Enum
 
 import arkindex.documents.dates
 import arkindex.documents.models
 import arkindex.project.fields
 import arkindex_common.enums
-import arkindex_common.ml_tool
 
 
 class Migration(migrations.Migration):
@@ -66,7 +66,7 @@ class Migration(migrations.Migration):
             name='DataSource',
             fields=[
                 ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
-                ('type', enumfields.fields.EnumField(enum=arkindex_common.ml_tool.MLToolType, max_length=10)),
+                ('type', enumfields.fields.EnumField(enum=Enum('MLToolType', ''), max_length=10)),
                 ('slug', models.CharField(max_length=100)),
                 ('name', models.CharField(max_length=100)),
                 ('revision', models.CharField(max_length=100)),
diff --git a/arkindex/documents/migrations/0013_datasource_type_length.py b/arkindex/documents/migrations/0013_datasource_type_length.py
index 93ff7fdd966d3062dcb1e8f9c6cd04a4b233cd72..d9b10a58b7003c7dc119244b83a1503ae9077bad 100644
--- a/arkindex/documents/migrations/0013_datasource_type_length.py
+++ b/arkindex/documents/migrations/0013_datasource_type_length.py
@@ -1,10 +1,10 @@
 # Generated by Django 2.2.10 on 2020-05-19 14:59
 
+from enum import Enum
+
 import enumfields.fields
 from django.db import migrations
 
-import arkindex_common.ml_tool
-
 
 class Migration(migrations.Migration):
 
@@ -16,6 +16,6 @@ class Migration(migrations.Migration):
         migrations.AlterField(
             model_name='datasource',
             name='type',
-            field=enumfields.fields.EnumField(enum=arkindex_common.ml_tool.MLToolType, max_length=50),
+            field=enumfields.fields.EnumField(enum=Enum('MLToolType', ''), max_length=50),
         ),
     ]