diff --git a/arkindex/documents/tests/test_datasource.py b/arkindex/documents/tests/test_datasource.py
index 3aee7866e08d8126fe8359d0f7830cfd74d7309c..b82c315706d6cb53491ec096462c45c24a06af33 100644
--- a/arkindex/documents/tests/test_datasource.py
+++ b/arkindex/documents/tests/test_datasource.py
@@ -1,19 +1,10 @@
-from unittest.mock import MagicMock
 from django.test import TestCase
-from arkindex_common.ml_tool import MLTool, MLToolType
+from arkindex_common.ml_tool import MLToolType
 from arkindex.documents.models import DataSource
 
 
 class TestDataSource(TestCase):
 
-    @classmethod
-    def setUpTestData(cls):
-        cls.ml_tool = MagicMock(spec=MLTool)
-        cls.ml_tool.type = MLToolType.Recognizer
-        cls.ml_tool.slug = 'muddy'
-        cls.ml_tool.name = 'Muddy Recognizer'
-        cls.ml_tool.version = '0.42'
-
     def test_str(self):
         ds = DataSource(
             type=MLToolType.Classifier,