diff --git a/python/packages/agbench/src/agbench/res/Dockerfile b/python/packages/agbench/src/agbench/res/Dockerfile
index aeea8fef9..a7da943f3 100644
--- a/python/packages/agbench/src/agbench/res/Dockerfile
+++ b/python/packages/agbench/src/agbench/res/Dockerfile
@@ -1,4 +1,4 @@
-FROM python:3.11
+FROM mcr.microsoft.com/devcontainers/python:3.11
 MAINTAINER AutoGen
 
 # Install packages
diff --git a/python/packages/autogen-studio/Dockerfile b/python/packages/autogen-studio/Dockerfile
index bb475db71..e8e9dbd81 100644
--- a/python/packages/autogen-studio/Dockerfile
+++ b/python/packages/autogen-studio/Dockerfile
@@ -1,4 +1,4 @@
-FROM python:3.10
+FROM mcr.microsoft.com/devcontainers/python:3.10
 
 WORKDIR /code