diff --git a/arkindex/training/tests/test_datasets_api.py b/arkindex/training/tests/test_datasets_api.py
index 83fc2ab45c3550b89e6aad1668fa5eea794fda5f..9d6da081461737c6c9a57f749abf814f8401f4ab 100644
--- a/arkindex/training/tests/test_datasets_api.py
+++ b/arkindex/training/tests/test_datasets_api.py
@@ -683,7 +683,7 @@ class TestDatasetsAPI(FixtureAPITestCase):
                 )
 
     def test_update_ponos_task_state_requires_dataset_in_process(self):
-        self.process.process_datasets.all().delete()
+        self.process.process_sets.all().delete()
         with self.assertNumQueries(5):
             response = self.client.put(
                 reverse("api:dataset-update", kwargs={"pk": self.dataset.pk}),
@@ -844,7 +844,7 @@ class TestDatasetsAPI(FixtureAPITestCase):
         self.assertEqual(self.dataset.state, DatasetState.Building)
 
     def test_partial_update_ponos_task_state_requires_dataset_in_process(self):
-        self.process.process_datasets.all().delete()
+        self.process.process_sets.all().delete()
         with self.assertNumQueries(5):
             response = self.client.patch(
                 reverse("api:dataset-update", kwargs={"pk": self.dataset.pk}),