diff --git a/evalharness/sandbox/__init__.py b/evalharness/sandbox/__init__.py
index 3f21ac3..9da3028 100644
--- a/evalharness/sandbox/__init__.py
+++ b/evalharness/sandbox/__init__.py
@@ -23,12 +23,12 @@ from .base import (
 )
 from .docker import DockerSandbox, docker_available, docker_serve, serve_env
 from .local import LocalSandbox
-from .prefetch import images_for_dataset, prefetch_images
+from .prefetch import images_for_dataset, images_for_samples, prefetch_images
 from .bg_prefetch import BackgroundPrefetcher
 
 __all__ = [
     'Sandbox', 'DockerSandbox', 'LocalSandbox', 'ExecResult', 'EnvHandle',
     'SANDBOX_REGISTRY', 'register_sandbox', 'get_sandbox', 'acquire', 'stop_all',
     'docker_serve', 'serve_env', 'docker_available',
-    'prefetch_images', 'images_for_dataset', 'BackgroundPrefetcher',
+    'prefetch_images', 'images_for_dataset', 'images_for_samples', 'BackgroundPrefetcher',
 ]