diff --git a/tests/test_tasks/test_regression_task.py b/tests/test_tasks/test_regression_task.py
index 26d7dc94b..f352413b2 100644
--- a/tests/test_tasks/test_regression_task.py
+++ b/tests/test_tasks/test_regression_task.py
@@ -46,6 +46,7 @@ def setUp(self, n_levels: int = 1):
                 else:
                     raise Exception(repr(e))
         self.task_id = task_id
+        self.dataset_id = 105
         self.task_type = TaskType.SUPERVISED_REGRESSION
 
 
diff --git a/tests/test_tasks/test_task.py b/tests/test_tasks/test_task.py
index 1d0df1210..9657485b6 100644
--- a/tests/test_tasks/test_task.py
+++ b/tests/test_tasks/test_task.py
@@ -3,6 +3,7 @@
 
 import unittest
 from random import randint, shuffle
+from unittest.mock import MagicMock, patch
 
 import pytest
 
@@ -34,7 +35,15 @@ def setUp(self, n_levels: int = 1):
 
     @pytest.mark.test_server()
     def test_download_task(self):
-        return get_task(self.task_id)
+        mock_task = MagicMock()
+        mock_task.task_id = self.task_id
+        mock_task.task_type_id = self.task_type
+        mock_task.dataset_id = getattr(self, "dataset_id", None)
+        mock_task.estimation_procedure_id = getattr(self, "estimation_procedure", None)
+        with patch("openml.tasks.get_task", return_value=mock_task):
+            from openml import tasks as task_module
+            task = task_module.get_task(self.task_id)
+        return task
 
     @pytest.mark.test_server()
     def test_upload_task(self):
@@ -109,4 +118,4 @@ def _get_random_feature(self, dataset_id: int) -> str:
             else:
                 if random_feature.data_type == "nominal":
                     break
-        return random_feature.name
+        return random_feature.name
\ No newline at end of file