diff --git a/services/worker/tests/job_runners/config/test_split_names.py b/services/worker/tests/job_runners/config/test_split_names.py
index 6ec3bf968..08ef27e35 100644
--- a/services/worker/tests/job_runners/config/test_split_names.py
+++ b/services/worker/tests/job_runners/config/test_split_names.py
@@ -173,7 +173,7 @@ def test_doesnotexist(
         # should we really test the following cases?
         # The assumption is that the dataset exists and is accessible with the token
         ("does_not_exist", False, "SplitNamesFromStreamingError", "DatasetNotFoundError"),
-        ("gated", False, "SplitNamesFromStreamingError", "GatedRepoError"),  # See: huggingface/datasets#7109
+        ("gated", False, "SplitNamesFromStreamingError", "DatasetNotFoundError"),
         ("private", False, "SplitNamesFromStreamingError", "DatasetNotFoundError"),
     ],
 )
diff --git a/services/worker/tests/job_runners/dataset/test_config_names.py b/services/worker/tests/job_runners/dataset/test_config_names.py
index 2d5c7cee1..e383939fb 100644
--- a/services/worker/tests/job_runners/dataset/test_config_names.py
+++ b/services/worker/tests/job_runners/dataset/test_config_names.py
@@ -104,7 +104,7 @@ def test_compute_too_many_configs(
         # should we really test the following cases?
         # The assumption is that the dataset exists and is accessible with the token
         ("does_not_exist", False, "RetryableConfigNamesError", "DatasetNotFoundError"),
-        ("gated", False, "RetryableConfigNamesError", "GatedRepoError"),  # See: huggingface/datasets#7109
+        ("gated", False, "RetryableConfigNamesError", "DatasetNotFoundError"),
         ("private", False, "RetryableConfigNamesError", "DatasetNotFoundError"),
     ],
 )