diff --git a/tests/unit/dataloader/test_torch_dataloader.py b/tests/unit/dataloader/test_torch_dataloader.py
index 61c8960b..2b771da8 100644
--- a/tests/unit/dataloader/test_torch_dataloader.py
+++ b/tests/unit/dataloader/test_torch_dataloader.py
@@ -420,7 +420,7 @@ def test_offsets():
                 assert feature_tensor == feature_result
 
 
-@pytest.mark.parametrize("device", ["cpu", 0, 1] if HAS_GPU else ["cpu"])
+@pytest.mark.parametrize("device", ["cpu", 0] if HAS_GPU else ["cpu"])
 def test_row_lengths_to_offsets_device(device):
     dataset = Dataset(make_df({"a": [1]}))
     loader = torch_dataloader.Loader(dataset, batch_size=1)