diff --git a/arkindex/training/tests/test_datasets_api.py b/arkindex/training/tests/test_datasets_api.py
index 16f3cfa09915f57a068ecac22060b566ef5ceb1c..4446928dd0ec6ab25e5e41b2f8ba3f5b89b25441 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}),