diff --git a/examples/mlperf/dataloader.py b/examples/mlperf/dataloader.py
index 04a7ef5991..6489143f5e 100644
--- a/examples/mlperf/dataloader.py
+++ b/examples/mlperf/dataloader.py
@@ -392,9 +392,9 @@ def batch_load_retinanet(dataset, val:bool, anchors:np.ndarray, base_dir:Path, b
       queue_in.put((idx, img, ann))
 
   def _setup_shared_mem(shm_name:str, size:Tuple[int, ...], dtype:dtypes) -> Tuple[shared_memory.SharedMemory, Tensor]:
-    if os.path.exists(f"/Users/flata/Downloads/shm/{shm_name}"): os.unlink(f"/Users/flata/Downloads/shm/{shm_name}")
+    if os.path.exists(f"/dev/shm/{shm_name}"): os.unlink(f"/dev/shm/{shm_name}")
     shm = shared_memory.SharedMemory(name=shm_name, create=True, size=prod(size))
-    shm_tensor = Tensor.empty(*size, dtype=dtype, device=f"disk:/Users/flata/Downloads/shm/{shm_name}")
+    shm_tensor = Tensor.empty(*size, dtype=dtype, device=f"disk:/dev/shm/{shm_name}")
     return shm, shm_tensor
 
   image_ids = sorted(dataset.imgs.keys())
diff --git a/test/external/external_test_datasets.py b/test/external/external_test_datasets.py
index a7045640c8..c14e268b86 100644
--- a/test/external/external_test_datasets.py
+++ b/test/external/external_test_datasets.py
@@ -130,9 +130,8 @@ class TestOpenImagesDataset(ExternalTestDatasets):
     for ((tinygrad_img, tinygrad_boxes, tinygrad_labels, _), (ref_img, ref_tgt)) in zip(tinygrad_dataloader, ref_dataloader):
       self._set_seed()
       ref_tgt = [ref_tgt]
-      print(f"{ref_img=} {ref_tgt=}")
 
-      ref_img, ref_tgt = transform(ref_img, [ref_tgt])
+      ref_img, ref_tgt = transform(ref_img.unsqueeze(0), ref_tgt)
       ref_tgt = postprocess_targets(ref_tgt, anchors.unsqueeze(0))
       ref_boxes, ref_labels = ref_tgt[0]["boxes"], ref_tgt[0]["labels"]