diff --git a/pkg/agentfs/examples/python.pip.Dockerfile b/pkg/agentfs/examples/python.pip.Dockerfile
index 129d3b8d..51cb5379 100644
--- a/pkg/agentfs/examples/python.pip.Dockerfile
+++ b/pkg/agentfs/examples/python.pip.Dockerfile
@@ -50,7 +50,7 @@ COPY . .
 # Pre-download any ML models or files the agent needs
 # This ensures the container is ready to run immediately without downloading
 # dependencies at runtime, which improves startup time and reliability
-RUN python "{{.ProgramMain}}" download-files
+RUN python -m livekit.agents download-files
 
 # --- Production stage ---
 # Build tools (gcc, g++, python3-dev) are not included in the final image
diff --git a/pkg/agentfs/examples/python.uv.Dockerfile b/pkg/agentfs/examples/python.uv.Dockerfile
index 403eb992..99738da0 100644
--- a/pkg/agentfs/examples/python.uv.Dockerfile
+++ b/pkg/agentfs/examples/python.uv.Dockerfile
@@ -54,7 +54,7 @@ COPY . .
 # Pre-download any ML models or files the agent needs
 # This ensures the container is ready to run immediately without downloading
 # dependencies at runtime, which improves startup time and reliability
-RUN uv run "{{.ProgramMain}}" download-files
+RUN uv run -m livekit.agents download-files
 
 # --- Production stage ---
 # Build tools (gcc, g++, python3-dev) are not included in the final image