diff --git a/open_ce/container_build.py b/open_ce/container_build.py
index f257560..8c38dbb 100644
--- a/open_ce/container_build.py
+++ b/open_ce/container_build.py
@@ -26,9 +26,9 @@
 from open_ce.inputs import Argument
 
 OPEN_CE_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), "."))
-BUILD_IMAGE_NAME = "builder"
+BUILD_IMAGE_NAME = "builder/ubi9"
 BUILD_IMAGE_PATH = os.path.join(OPEN_CE_PATH, "images", BUILD_IMAGE_NAME)
-BUILD_CUDA_IMAGE_NAME = "builder-cuda"
+BUILD_CUDA_IMAGE_NAME = "builder-cuda/ubi9"
 BUILD_CUDA_IMAGE_PATH = os.path.join(OPEN_CE_PATH, "images", BUILD_CUDA_IMAGE_NAME)
 LOCAL_FILES_PATH = os.path.join(os.path.join(os.getcwd(), "local_files"))