diff --git a/tests/beignet/datasets/test__uniref_dataset.py b/tests/beignet/datasets/test__uniref_dataset.py
index 391dd916ba..2037daa601 100644
--- a/tests/beignet/datasets/test__uniref_dataset.py
+++ b/tests/beignet/datasets/test__uniref_dataset.py
@@ -4,8 +4,9 @@
 
 
 class TestUnirefDataset:
-    @mock.patch("beignet.io.download_and_extract_archive", return_value=True)
-    def test_init(self, download_func):
+    @mock.patch("beignet.io._download.download_and_extract_archive", return_value=True)
+    @mock.patch("beignet.datasets.UniRef50Dataset.__init__", return_value=None)
+    def test_init(self, download_func, uniref_init):
         dataset = UniRef50Dataset(root="/tmp/data", download=True)
 
         assert download_func.called_once()