diff --git a/tests/datasets/test_generic.py b/tests/datasets/test_generic.py
index 09a5c2f..7f72bfe 100644
--- a/tests/datasets/test_generic.py
+++ b/tests/datasets/test_generic.py
@@ -67,8 +67,8 @@ def test_add(self):
         dataset1 = DatasetArray(name="Array", root=raw_path, download=False)
         dataset2 = DatasetArray(name="Array", root=raw_path, download=False)
 
-        dataset1.load()
-        dataset2.load()
+        dataset1._load_cpu()
+        dataset2._load_cpu()
 
         np1 = np.load(raw_path)
         np2 = np.load(raw_path)
@@ -87,8 +87,8 @@ def test_sub(self):
         dataset1 = DatasetArray(name="Array", root=raw_path, download=False)
         dataset2 = DatasetArray(name="Array", root=raw_path, download=False)
 
-        dataset1.load()
-        dataset2.load()
+        dataset1._load_cpu()
+        dataset2._load_cpu()
 
         np1 = np.load(raw_path)
         np2 = np.load(raw_path)
@@ -107,8 +107,8 @@ def test_mul(self):
         dataset1 = DatasetArray(name="Array", root=raw_path, download=False)
         dataset2 = DatasetArray(name="Array", root=raw_path, download=False)
 
-        dataset1.load()
-        dataset2.load()
+        dataset1._load_cpu()
+        dataset2._load_cpu()
 
         np1 = np.load(raw_path)
         np2 = np.load(raw_path)