diff --git a/viscy/data/combined.py b/viscy/data/combined.py
index 28cf9339..87036723 100644
--- a/viscy/data/combined.py
+++ b/viscy/data/combined.py
@@ -7,7 +7,7 @@
 from torch.utils.data import ConcatDataset, DataLoader, Dataset
 
 from viscy.data.distributed import ShardedDistributedSampler
-from viscy.data.hcs_ram import _collate_samples
+from viscy.data.hcs import _collate_samples
 
 
 class CombineMode(Enum):